树形DP hdu1520

#include <iostream>
#include <cstdio>

using namespace std;

struct node
{
    int fro,to,next;
}e[36000000];
int dp[6000][2];
int num=0;
int head[6000];
int val[6000];
bool isye[6000];

void addEdge(int _fro,int _to)
{
    num++;
    e[num].fro=_fro;
    e[num].to=_to;
    e[num].next=head[_fro];
    head[_fro]=num;
}

int dfs(int x)
{
    dp[x][0]=0;
    dp[x][1]=val[x];
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        dfs(v);
        dp[x][0]+=max(dp[v][0],dp[v][1]);
        dp[x][1]+=dp[v][0];
    }
    return max(dp[x][0],dp[x][1]);
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&val[i]);
        }
        int a,b;
        for(int i=0;i<=n;i++)
        {
            head[i]=-1;
            isye[i]=true;
        }
        while(scanf("%d%d",&a,&b)!=EOF)
        {
            if(a==0&&b==0)
                break;
            addEdge(b,a);
            isye[a]=false;
        }
        for(int i=1;i<=n;i++)
        {
            if(isye[i])
            {
                cout<<dfs(i)<<endl;
            }
        }
    }
    return 0;
}