【CODEVS1380】没有上司的舞会
Description
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
Input
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。
Output
输出最大的快乐指数。
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
HINT
各个测试点1s
题解
树形dp,f[i][0]表示i不参加时他和他的下属制造的的最大欢乐度,f[i][0]表示i参加是他和他的下属制造的的最大欢乐度。
给i和他的下属连一条边,如果i参加,那i的下属只能不参加,如果i不参加,i的下属可以选择参加或者不参加
边表存储。
#include<iostream> #include<cstdio> #include<cstring> using namespace std; struct node{ int to,next; }e[6001]; int f[6001][2]; int head[6001],a[6001],r[6001]; int cnt,n; void ins(int u,int v){ e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt; } void dp(int x){ f[x][0]=0;f[x][1]=a[x]; for (int i=head[x];i;i=e[i].next){ int v=e[i].to; dp(v); f[x][0]+=max(f[v][0],f[v][1]); f[x][1]+=f[v][0]; } } int main(){ scanf("%d",&n); for (int i=1;i<=n;i++) scanf("%d",&a[i]); for (int u=1,v=1;u&&v;){ scanf("%d%d",&u,&v); ins(v,u); r[u]=1; } for (int i=1;i<=n;i++) if (!r[i]){ dp(i); printf("%d",max(f[i][0],f[i][1])); break; } }