POJ Anniversary party 树形DP

/*
树形dp：
给一颗树，要求一组节点，节点之间没有父子关系，并且使得所有的节点的权值和最大
对于每一个节点，我们有两种状态
dp[i][0]表示不选择节点i，以节点i为根的子树所能形成的节点集所能获得的最大权值和
dp[i][1]表示选择节点i ，同上！

转移方程：
dp[i][0]+=max(dp[i_son][1],dp[i_son][0])如果没选择的话，那么子树可选择可不选择
dp[i][1]+=dp[i_son][0] 选择了之后，子树只能不选择
最后输出max(dp[i][0],dp[i][1])即可
*/ 

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define maxn 6002
vector<int> v[maxn];
bool b[maxn];
int dp[maxn][2],hp[maxn];
void dfs(int k)
{
    int i,len=v[k].size();
    dp[k][0]=0 ;
    dp[k][1]=hp[k];
    if(len==0)
    return ;
    for(i=0;i<len;i++)
    {
        dfs(v[k][i]);
        dp[k][0]+=max(dp[v[k][i]][1],dp[v[k][i]][0]);
        dp[k][1]+=dp[v[k][i]][0];
    }
}
int main()
{
    int i,n;
    int k,l;
    while(cin>>n)
    {
        for(i=1;i<=n;i++)
        cin>>hp[i],v[i].clear();
        memset(b,0,sizeof(b));
        while(cin>>l>>k&&k||l)
        {
            v[k].push_back(l);
            b[l]=1;
        }
        for(i=1;i<=n;i++)
        {
            if(!b[i])
                break;
        }
        b[i]=1;
        dfs(i);
        cout<<max(dp[i][1],dp[i][0])<<endl;
    }
    return 0;
}