HDU 1520 Anniversary party

题目大意：一个公司里有上下级之分，每一个在公司里的人都有一个快乐指数。有一天要举办一个聚会，人们不想自己的直系上司和自己一起出席。求出最大的开心指数。

思路：公司里的上司关系组成了一棵树，就是在树上的DP。

#include <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
int dp[6005][2];    //第二维的1代表选自己   0代表不选自己
int w[6005];
vector <int> child[6005];   //记录儿子

int n;
int v[6005];

void dfs(int x)
{
    for(int i=0;i<child[x].size();i++)
    {
        int c=child[x][i];   
        dfs(c);
        dp[x][1]+=dp[c][0];     //选自己就不能选下属  下属不乐意。。。
        dp[x][0]+=max(dp[c][1],dp[c][0]);   //不选自己就考虑一下下属要不要。 = =
    }
}


int main()
{

    while(scanf("%d",&n)!=EOF)
    {
        memset(v,0,sizeof(v));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            child[i].clear();
            scanf("%d",&dp[i][1]);   
        }
        int a, b;
        while(scanf("%d%d",&a,&b)!=EOF && (a && b))
        {
            v[a]=1;    //把下属都标记   最后没标记的就是最高上司   就是根节点
            child[b].push_back(a);
        }
        int i;
        for(i=1;i<=n;i++)
        {
            if(v[i]==0)
            {
                dfs(i);   //找到了根节点
                break;
            }
        }

        printf("%d\n",max(dp[i][0],dp[i][1]));    
    }

    return 0;
}