没有上司的舞会
1380 没有上司的舞会
时间限制: 1 s
空间限制: 128000 KB
题目描述 Description
Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。
输入描述 Input Description
第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。
输出描述 Output Description
输出最大的快乐指数。
样例输入 Sample Input
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
样例输出 Sample Output
5
·树形DP
F[i][0]表示i没来的最大快乐指数，F[i][1]表示i来了的快乐指数。
从子树转移即可。
code：
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int n,fa[],root,f[][];
vector <int> a[];
void dp(int node){
for(int i=;i<a[node].size();i++){
int to=a[node][i];
dp(a[node][i]);
f[node][]+=max(f[to][],f[to][]);
f[node][]+=f[to][];
}
}
int main(){
scanf("%d",&n);
for(int i=;i<=n;i++)scanf("%d",&f[i][]);
for(int i=;i<=n;i++)fa[i]=i;
while(){
int x,y;scanf("%d%d",&x,&y);
if(!x&&!y)break;
a[y].push_back(x);
fa[x]=y;
}
for(int i=;i<=n;i++)
if(fa[i]==i){root=i;break;}
dp(root);
printf("%d",max(f[root][],f[root][]));
return ;
}