题目连接: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;
}