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;
}