P1352 没有上司的舞会(树形DP)
题目描述
某大学有 nn 个职员,编号为 1\ldots n1…n。
他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。
现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 r_iri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。
所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入格式
输入的第一行是一个整数 nn。
第 22 到第 (n + 1)(n+1) 行,每行一个整数,第 (i+1)(i+1) 行的整数表示 ii 号职员的快乐指数 r_iri。
第 (n + 2)(n+2) 到第 2n2n 行,每行输入一对整数 l, kl,k,代表 kk 是 ll 的直接上司。
输出格式
输出一行一个整数代表最大的快乐指数。
#include<bits/stdc++.h> using namespace std; const int maxn=2e5+100; int n; vector<int> g[maxn]; int dp[2][maxn]; //dp[0]表示不选择该节点的最优解 //dp[1]表示选择该节点的最优解 int a[maxn]; void dfs (int u,int pre) { dp[0][u]=0; dp[1][u]=a[u]; for (int v:g[u]) { if (v==pre) continue; dfs(v,u); dp[0][u]+=max(dp[1][v],dp[0][v]); dp[1][u]+=dp[0][v]; } } int main () { scanf("%d",&n); for (int i=1;i<=n;i++) scanf("%d",a+i); for (int i=1;i<n;i++) { int x,y; scanf("%d%d",&x,&y); g[x].push_back(y); g[y].push_back(x); } dfs(1,0); printf("%d\n",max(dp[0][1],dp[1][1])); }