196、【动态规划】AcWing —— 285. 没有上司的舞会(C++版本)
创始人
2024-06-01 10:48:09
0

题目描述

在这里插入图片描述
在这里插入图片描述
原题链接:285. 没有上司的舞会

解题思路

题目是要构造一个树形关系,在选择时,有直接上下级关系的不能选,但是对于有间接上下级关系的和非上下级关系的都可以选。

  • 动态规划五步曲:

(1)dp[i][2]含义: dp[i][0]:以i为根节点并且不选节点i时,具有的最大快乐数。dp[i][1]:以i为根节点并且选节点i时,具有的最大快乐数。

(2)递推公式: dp[i][1]=dp[i][1]+dp[j][0]dp[i][1] = dp[i][1] + dp[j][0]dp[i][1]=dp[i][1]+dp[j][0](其中j为i的孩子节点),意为选择当前节点的情况下加上不选择孩子节点的情况;dp[i][0]=max(dp[j][0],dp[j][1])dp[i][0] = max(dp[j][0], dp[j][1])dp[i][0]=max(dp[j][0],dp[j][1]),意为在不选择i节点的情况下,选择孩子节点和不选择孩子节点时候找到一个最大值。

(3)dp数组初始化: dp[i][1] = happy[i],当选择孩子节点时,初始时候为自身的快乐数值。

(4)遍历顺序: 后序遍历,从左到右,从下到上,将低层结果传递给上一层。

(5)举例: (省略)

#include 
#include 
#include using namespace std;const int N = 6010;
int n;
int happy[N];
int dp[N][2];
int e[N], ne[N], h[N], idx;     //邻接表。e[i]:存放新节点,ne[i]:新节点的next,h[i]:头指针,idx:指向全局的下标
bool has_father[N];             // 判断下表i对应的节点是否有父节点。void add(int a, int b) {        // 头插法,将b插入到a中,a为父节点,b为子节点e[idx] = b, ne[idx] = h[a], h[a] = idx++;       // 构建节点b,让b的next指向a的next,让a的next指向b,更新idx。
}void dfs(int u) {dp[u][1] = happy[u];                        // 初始化构建快乐数for(int i = h[u]; i != -1; i = ne[i]) {     // 从树的根节点开始,遍历它下一层的所有孩子节点int j = e[i];                           // 找到节点u的孩子节点dfs(j);                                 // 采用后序遍历,先向下回溯遍历,再计算结果,将结果结果传递给上一层// 注意:这里加的时候要加到最初的起始节点处dp[u][0] += max(dp[j][1], dp[j][0]);    // 第一种情况:不选u,则可能最大值为选择孩子或不选孩子dp[u][1] += dp[j][0];                   // 第二种情况:选择当前节点,则就不选孩子节点(注意此时可选孙子节点)。}
}int main() {// 1、输入数据,构建关系树cin >> n;for(int i = 1; i <= n; i++)         cin >> happy[i];memset(h, -1, sizeof h);for(int i = 0; i < n - 1; i++) {int b, a;cin >> b >> a;              // b为下级,a为直接上司has_father[b] = true;       // b存在父节点add(a, b);                  // 构建a为b的父节点关系结构}// 2、寻找根节点int root = 1;while(has_father[root] == true)     root++;// 3、找到最大快乐数dfs(root);cout << max(dp[root][0], dp[root][1]) << endl;return 0;
}

参考视频:9.80 树形DP 没有上司的舞会——信息学竞赛培训课程

相关内容

热门资讯

激发暑期文旅消费新动能   内蒙古阿尔山,漫步国家森林公园,感受清凉夏日;甘肃敦煌,坐在鸣沙山上,听一场星空演唱会;江苏连云...
日股收跌0.19% 格隆汇7月11日|日经225指数收盘下跌76.68点,跌幅0.19%,报39569.68点。
遵义小伙跳水救人牺牲,年仅31... (转自:遵义晚报)7月8日下午6时,遵义市播州区出现感人一幕——尚嵇镇大坝社区全军坝组28岁的村民陈...
“小记者”走进河北省邯郸市食品... 中国质量新闻网讯 7月7日下午,河北省邯郸市食品药品检验中心(市食品药品安全科普基地)联合邯郸新闻传...
局地超42℃!下周河南持续高温... 大范围高温闷热天气强势来袭!12日起,西太平洋副热带高压将再次控制我省,受其影响,13日至20日我省...
毕市监处罚[2025]1021... 毕市监处罚[2025]1021号浏览:1次基本信息:处罚机关毕节市市场监督管理局处罚日期2025年0...
达势股份-达美乐中国以4D战略...   达势股份-达美乐中国("达美乐中国"或"达势股份"或"公司")(1405.HK)是达美乐比萨在中...
“向上取整”:变相“明抢” 转自:中国质量报□ 胡立彪近日,有媒体对快递行业称重计费问题进行测评,结果发现,被测评的8家快递企业...
以改革破局 盐城经开区多点突破... 中新网江苏新闻7月11日电(倪玲)改革是发展的根本动力,开放是进步的必由之路。近年来,盐城经开区将推...
中国代表强调国际刑事法院应依法... 转自:千龙网新华社联合国7月10日电 中国常驻联合国副代表孙磊10日在安理会审议国际刑事法院涉苏丹问...