P1352 没有上司的舞会
![[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-rnF5OO5k-1591067870768)(D:\坚果同步文件夹\我的坚果云\算法\图库\1589599936409.png)]](https://i-blog.csdnimg.cn/blog_migrate/23ef668607d6737685394949c9ccd910.png)
![[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-qe0iv5je-1591067870770)(D:\坚果同步文件夹\我的坚果云\算法\图库\1589599949942.png)]](https://i-blog.csdnimg.cn/blog_migrate/29d1554e93ebfc0f2421fa25025beadc.png)
题意
- 一家公司有n个员工,编号为1~n。
- 他们的关系就像一棵以校长为根的数,父节点就是子节点的直接上司。
- 每个员工都有一个快乐指数。
- 现在要开一个周年庆典,没有员工愿意和直接上司参加舞会,问怎样安排能让快乐值最大,求最大值。
- 数据范围:N≤6000
想法
- DFS+DP,dp[x,1]表示x去,dp[x,0]表示x不去
- dp[x,1]:=sum(dp[所有子节点,0])+快乐值[x]; //x去了,下属不能去,所以dp[所有子节点,0]
- dp[x,0]:=sum(max(dp[所有子节点,0],dp[所有子节点,1])); //x不去,下属可去可不去,所以max(dp[所有子节点,0],dp[所有子节点,1])
- 注意先求下属,再求上司。
- 利用链式前向星保存树
#include <iostream>
using namespace std;
const int maxn = 6e3+10;
int n,r[maxn],head[maxn];
int cnt;
int d[maxn];
int dp[maxn][2];
struct edge{
int to,w,next;
}edges[maxn];
void addEgde(int u,int v){
edges[++cnt].next = head[u];
edges[cnt].to = v;
head[u] = cnt;
}
void dfs(int root){
dp[root][1] = r[root];
dp[root][0] = 0;
for (int i = head[root]; i!=0 ; i=edges[i].next) {
int j = edges[i].to;
dfs(j);
dp[root][0] += max(dp[j][1],dp[j][0]);
dp[root][1] += dp[j][0];
}
}
int main(){
cin >> n;
for (int i = 1; i <= n; ++i) {
cin >> r[i];
}
for (int i = 1; i <= n-1; ++i) {
int x,y;
cin >> x >> y;
d[x]++;
addEgde(y,x);
}
int root = 0;
for (int i = 1; i <= n; ++i) {
if(d[i] == 0){
root = i;
break;
}
}
dfs(root);
cout << max(dp[root][1],dp[root][0]);
}