hdu1520 树形dp

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=6005;

struct node
{
    int v;
    int next;
} node[N<<1];

int dp[N][2];
int a[N],head[N],in[N];
int k;

void addedge(int u,int v)  ///u是父亲，v是孩子
{
    node[k].v=v;
    node[k].next=head[u];
    head[u]=k++;
}

int dfs(int u)
{
    dp[u][0]=0;
    dp[u][1]=a[u];
    for(int i=head[u]; i!=-1; i=node[i].next)
    {
        int v=node[i].v;
        dfs(v);
        dp[u][0]+=max(dp[v][1],dp[v][0]);
        dp[u][1]+=dp[v][0];
    }
    return max(dp[u][1],dp[u][0]);
}

int main()
{
    int n,x,y;
    while(scanf("%d",&n)==1)
    {
        k=0;
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        memset(in,0,sizeof(in));
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        while(scanf("%d%d",&x,&y) && (x+y))
        {
            addedge(y,x);
            in[x]++;
        }
        int ans=0;
        for(int i=1; i<=n; i++)  ///其实这是一个森林，找到每棵树的根，求和
            if(!in[i]) ans+=dfs(i);
        printf("%d\n",ans);
    }
    return 0;
}