HDOJ 1520 Anniversary party（树DP）

题意：求树的最大点独立集中点的数目。

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 6001
int n,e;
int w[N],first[N],next[N],v[N],p[N];
int dp[N][2];
void init()
{
    e=0;
    memset(first,-1,sizeof(first));
    memset(p,0,sizeof(p));
    memset(dp,0,sizeof(dp));
}
void add(int a,int b)
{
    v[e]=b;
    next[e]=first[a];
    first[a]=e++;
}
void dfs(int a)
{
    dp[a][0]=0;
    dp[a][1]=w[a];
    int i,b;
    for(i=first[a];~i;i=next[i])
    {
        b=v[i];
        if(b==p[a]) continue;
        dfs(b);
        dp[a][0]+=max(dp[b][0],dp[b][1]);
        dp[a][1]+=dp[b][0];
    }
}
int main()
{
    int a,b;
    while(~scanf("%d",&n))
    {
        init();
        for(int i=1;i<=n;i++)   scanf("%d",&w[i]);
        do
        {
            scanf("%d%d",&a,&b);
            if(!p[a])
            {
                p[a]=b;
                add(b,a);
            }
        }while(a|b);
        for(int i=1;i<=n;i++)
        {
            if(p[i]==0)
            {
                dfs(i);
                printf("%d\n",max(dp[i][0],dp[i][1]));
                break;
            }
        }
    }
    return 0;
}