hdu_Anniversary party_(树形DP入门题)

来源:互联网 发布:java反序列化利用 ma 编辑:程序博客网 时间:2024/06/05 09:15

题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意:有N个人,N-1个人有自己的上司,每个人有一个快乐值,如果这个人参加了聚会,那么这个人的直接上司将不嫩参加,问最大的快乐值为多少

题解:入门的树形DP题,dp[i][0]表示第i个人不去,dp[i][1]表示第i个人去。

#include<cstdio>#define FFC(i,a,b) for(int i=a;i<=b;i++)#define max(a,b) ((a)>(b)?(a):(b))const int maxn=6010;int n,c[maxn],v[maxn],g[maxn],nxt[maxn],ed,xx,yy,ik,f[maxn],root,dp[maxn][2];void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}void fuck(int now){dp[now][1]=c[now];for(int i=g[now];i;i=nxt[i]){fuck(v[i]);dp[now][0]+=max(dp[v[i]][1],dp[v[i]][0]);dp[now][1]+=dp[v[i]][0];}}int main(){while(~scanf("%d",&n)){FFC(i,1,n)scanf("%d",&c[i]);for(ik=ed=0,root=1;ik<=n;ik++)g[ik]=0,f[ik]=ik;while(scanf("%d%d",&xx,&yy),xx+yy)adg(yy,xx),f[xx]=yy;FFC(i,1,n)dp[i][0]=dp[i][1]=0;while(f[root]!=root)root=f[root];fuck(root);printf("%d\n",max(dp[root][1],dp[root][0]));}return 0;}


0 0
原创粉丝点击