POJ 2342 && HDU 1520 Anniversary party（树形DP）

Description
话说一个公司的一些然要去参加一个party，每个人有一个愉悦值，而如果某个人的直接上司在场的话会非常扫兴，所以避免这样的安排，问给出n个人，每个人的愉悦值以及他们的上司所属关系，问你让那些人去可以让总的愉悦值最大，并求出这个值
Input
第一行为总人数n，之后n行为每个人的愉悦值，然后每行两个整数a和b表示所属关系b是a的上司，以0 0结束所属关系输入
Output
输出最大总愉悦值
Sample Input
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
Sample Output
5
Solution
这个公司的人事关系可以根据给出的数据得到一个树，最上面的是最高层，往下依次，我们要做的就是在树的节点处进行dp。
用dp[i][0]表示当前i这个人不去，dp[i][1]表示当前i这个人去
那么dp[fa[i]][1]+=dp[i][0]
而 dp[fa[i]][0]+=max(dp[i][0],dp[i][1])
Code

#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define maxn 6666
int a[maxn],fa[maxn],dp[maxn][2];
vector<int> vec[maxn];
int n;
void dfs(int root)
{
    dp[root][1]=a[root];
    int len=vec[root].size();
    for(int i=0;i<len;i++)
        dfs(vec[root][i]);
    for(int i=0;i<len;i++)
    {
        dp[root][0]+=max(dp[vec[root][i]][1],dp[vec[root][i]][0]);//这个人不去 
        dp[root][1]+=dp[vec[root][i]][0];//这个人去 
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            //初始化 
            vec[i].clear(); 
            fa[i]=-1;
            dp[i][0]=dp[i][1]=0;
        }
        int u,v;
        while(scanf("%d%d",&u,&v),u||v)
        {
            vec[v].push_back(u);
            fa[u]=v;
        }
        int temp=1;
        while(fa[temp]!=-1)//找到根节点 
            temp=fa[temp];
        dfs(temp);//从根节点开始搜索 
        int ans=max(dp[temp][0],dp[temp][1]);
        printf("%d\n",ans);
    }
    return 0;
}