hdu1520Anniversary party 树形dp水题

/*
dp[i][0]表示第i个人不去的时候能得到的最大值
dp[i][1]表示第i个人去的时候得到的最大值
状态转移方程：
dp[i][0]+=max(dp[next][0],dp[next][1])
dp[i][1]+=dp[next][0]
其中next为其子节点
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define maxn 6010
vector<int> vec[maxn];
int dp[maxn][2];
int value[maxn];
void dfs(int root)
{
    int i;
    for(i=0;i<vec[root].size();i++)
    {
        int next=vec[root][i];
        dfs(next);
        dp[root][0]+=max(dp[next][0],dp[next][1]);
        dp[root][1]+=dp[next][0];
    }
}
int main()
{
    int N;
    while(scanf("%d",&N)!=EOF)
    {
        int i,j;
        memset(dp,0,sizeof(dp));
        for(i=1;i<=N;i++)
        scanf("%d",&dp[i][1]);
        int l,k;
        for(i=0;i<=N;i++)
        vec[i].clear();
        int temp[maxn];
        int root;
        memset(temp,0,sizeof(temp));
        while(scanf("%d%d",&l,&k),l||k)
        {
                vec[k].push_back(l);
                temp[l]=1;
        }
        for(i=1;i<=N;i++)
        if(!temp[i])
        root=i;
        dfs(root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}