hdu_Anniversary party_(树形DP入门题)

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