hdu1520，树形dp

最近开始搞树形dp，结果搞的第一题，开始题目看错，后来又因为数组开小，结果wa了许多次，哭~~~不是树么？存边的数组开得和点一样大，居然存不下。。。

代码如下：

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#define INF 200000000
int dp[6005][2],first[6005],next[12050],head[6005],a[6005],num[6005][2];
struct edge
{
    int u;
    int v;
}en[12050];
int max(int a,int b)
{
    return a>b?a:b;
}
void getdp(int h)
{
    int i,j,k,l;
    if(first[h]==-1)
    {
            dp[h][0]=0;
            dp[h][1]=a[h];
    }
    else
    {
        for(i=first[h];i!=-1;i=next[i])
        {
                getdp(en[i].v);
                dp[h][1]+=dp[en[i].v][0];
                dp[h][0]+=max(dp[en[i].v][0],dp[en[i].v][1]);
        }
        dp[h][1]+=a[h];
    }
}
int main()
{
    int n,i,j,k,l,le,ri,idx,s,f,h;
    while(scanf("%d",&n)!=EOF)
    {
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
    idx=0;
    memset(head,-1,sizeof(head));
    memset(first,-1,sizeof(first));
    memset(next,-1,sizeof(next));
    memset(dp,0,sizeof(dp));
    while(scanf("%d%d",&s,&f)&&(s||f))
    {
        en[++idx].u=f;
        en[++idx].v=s;
        head[s]=f;
        next[idx]=first[f];
        first[f]=idx;
    }
    i=1;
    while(head[i]!=-1)
    {
        i=head[i];
    }
    h=i;
    getdp(h);
    printf("%d\n",(dp[h][0]>dp[h][1]?dp[h][0]:dp[h][1]));
    }
    return 0;
}