HDU 1520 Anniversary party 树形DP

有一个party，为了避免不欢乐，员工与他的直接上司不能同时参加，现给你员工之间的关系图，还有每个员工参加party的欢乐值，问你得到的最大欢乐值是多少。

如果贪心，那么策略是······想着想着就变成了DP，对于每一个节点，有取与不取两种状态，那么转移方程是：

dp[ i ][ 0 ]=sum{ max { dp[ j ] [ 0 ] , dp [ j ] [ 1 ] } }//如果不取这个点，那么它的孩子可取也可不取，选最优

dp[ i ] [ 1 ] = sum{ dp [ j ] [ 0 ] };//如果取这个点，那么它的孩子只能不取

#include<iostream>
#include<queue>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
struct node
{
    int y,next;
}ed[6010];
int h[6010],dp[6010][3],cnt[6010],v[6010];
int n,nn,p;
void init(int i,int j)
{
    ed[p].y=i;
    ed[p].next=h[j];
    h[j]=p++;
}
void dfs(int now)
{
    int i;
    if (!h[now])
    {
        dp[now][0]=0;
        dp[now][1]=v[now];
        return ;
    }
    dp[now][1]+=v[now];
    i=h[now];
    while (i)
    {
        dfs(ed[i].y);
        dp[now][0]+=(dp[ed[i].y][0]>dp[ed[i].y][1]?dp[ed[i].y][0]:dp[ed[i].y][1]);
        dp[now][1]+=dp[ed[i].y][0];
        i=ed[i].next;
    }
}
int main()
{
    int root,i,j,k;
    while (scanf("%d",&n)!=EOF)
    {
        p=1;
        memset(h,0,sizeof(h));
        memset(dp,0,sizeof(dp));
        memset(ed,0,sizeof(ed));
        memset(cnt,0,sizeof(cnt));
        for (i=1;i<=n;i++)
        {
            scanf("%d",&v[i]);
        }
        while (scanf("%d%d",&j,&k)&&(j||k))
        {
            init(j,k);
            cnt[j]++;
        }
        for (i=1;i<=n;i++)
            if (cnt[i]==0)
            {
                dfs(i);break;
            }
        printf("%d\n",(dp[i][0]>dp[i][1])?dp[i][0]:dp[i][1]);
    }
    return 0;
}