HDU 1520 Anniversary party(树形DP)

http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意：n个点构成一棵树，每个点都有一个权值，从中选出一些点，使权值之和最大，所选择的点不能直接相连，即不能是父子关系

算是树形DP的入门题吧

dp[i][0]表示不选i点，dp[i][1]表示选择该点

dp[i][0]=sum(max(dp[j][0],dp[j][1]))  j是i的子节点

dp[i][1]=sum(dp[j][0])+val[i];  

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define nMAX 6600
using namespace std;
int val[nMAX],dp[nMAX][2],head[nMAX],s_edge;
struct Edge
{
    int v,nxt;
}edge[nMAX];
bool in[nMAX],out[nMAX];
int n;
int max(int a,int b)
{
    return a>b?a:b;
}
void addedge(int u,int v)
{
    s_edge++;
    edge[s_edge].v=v;
    edge[s_edge].nxt=head[u];
    head[u]=s_edge;
}
void dfs(int u)
{
    if(!out[u]) {dp[u][0]=0; dp[u][1]=val[u]; return ;}
    int SUM=0,unSUM=0;
    for(int e=head[u];e;e=edge[e].nxt)
    {
        int v=edge[e].v;
        dfs(v);
        unSUM+=max(dp[v][0],dp[v][1]);
        SUM+=dp[v][0];
    }
    dp[u][0]=unSUM;
    dp[u][1]=SUM+val[u];
}
int main()
{
    int i,j,k;
    while(~scanf("%d",&n))
    {
        for(i=1;i<=n;i++)   scanf("%d",&val[i]);
        memset(head,0,sizeof(head));
        memset(in,0,sizeof(in));
        memset(out,0,sizeof(out));
        s_edge=0;
        while(~scanf("%d%d",&j,&k))
        {
            if(j==0&&k==0) break;
            out[k]=1;
            in[j]=1;
            addedge(k,j);
        }
        for(i=1;i<=n;i++)
            if(!in[i]) break;
        dfs(i);
        int ans=max(dp[i][0],dp[i][1]);
        printf("%d\n",ans);
    }
    return 0;
}