原题链接:没有上司的晚会
解题思路:
这是一道树形DP,可以用深搜和动规来实现
定义dp数组:
int dp[10010][3];//dp[x][0]代表x没来快乐最大值,dp[x][1]代表x来了快乐最大值
定义一个二维数组,储存一个节点的下属
vector g[10010];
深搜主体代码:
void dfs(int x) //当前节点 { dp[x][0] = 0; dp[x][1] = v[x];//初始化 for(int i = 0; i < g[x].size(); i++) { dfs(g[x][i]);//先深搜!!搜索完才可以有数据来递推 dp[x][0] += max(dp[g[x][i]][0],dp[g[x][i]][1]);//上司没来取最大值 dp[x][1] += dp[g[x][i]][0];//上司来了下属只能不来 } }
还有一个问题,从哪里开始搜索?得找到根节点
根节点没有父节点
用b数组来存储x是否有父节点
int v[10010];
在输入过程中,顺便标记b数组
输入结束后,再遍历找出根节点root
void find_root() { for(int i = 1; i <= n; i++) if(b[i] == false) { root = i; break; } }
最后输出root节点的最大值
cout << max(dp[root][0],dp[root][1]) << endl;
参考代码:
#include <bits/stdc++.h> using namespace std; int dp[10010][3]; int v[10010]; vector g[10010]; bool b[10010]; int n,root; void read() { memset(b,false,sizeof(b)); cin >> n; for(int i = 1; i > v[i]; int x,y; while(1) { cin >> x >> y; if(!x && !y) break; g[y].push_back(x); b[x] = true; } } void find_root() { for(int i = 1; i <= n; i++) if(b[i] == false) { root = i; break; } } void dfs(int x) { dp[x][0] = 0; dp[x][1] = v[x]; for(int i = 0; i < g[x].size(); i++) { dfs(g[x][i]); dp[x][0] += max(dp[g[x][i]][0],dp[g[x][i]][1]); dp[x][1] += dp[g[x][i]][0]; } } int main() { read(); find_root(); dfs(root); cout << max(dp[root][0],dp[root][1]) << endl; return 0; }
0.0分
1 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复