树形DP

Anniversary party
要有一个庆祝乌拉尔州立大学80周年。这所大学有一个员工的层次结构。这意味着主管的关系形成一个树在特列季亚科夫校长”。为了使党的滑稽的每一个，校长不希望员工和他或她的直属主管到场。人事办公室评估每个员工的欢乐，所以每个人都有一些数（等级）连接到他或她。你的任务是与客人欢宴评级的最大金额列出客人。
输入
员工编号从1到N的第一行输入包含一个数N = N = 6 < 1 < 000。接下来的N行包含相应员工的欢乐度。欢乐的评级是一个整数的范围从128到127 -。之后去行，描述一个主管关系树。每行树规范的形式：

L K

这意味着，第k个员工是l个员工的直接主管。输入用行结束。

0 0
输出
输出应包含顾客评分之和最大。

#include<cstdio>
#include<cstring>
#include<vector> 
#include<algorithm>
using namespace std;//一个节点有两个状态，所以两个状态的最大值都要记录一下。0和1， 

const int maxn = 6000 + 10;
int n;
//int a[maxn];
int dp[maxn][2];//第二个位置存状态（0或1），dp记录的是目前为0或为1最大的值 
vector<int> son[maxn];                  //开二维很容易爆空间 
int fa[maxn]; 
int vis[maxn];

void solve(int root){
    vis[root] = 1;
    for(int i = 0;i<son[root].size();i++){//遍历所有的子节点 
        int temp = son[root][i];//记录一下此时的子节点 
        if(!vis[temp]){
            solve(temp);
            dp[root][1] += dp[temp][0];
            dp[root][0] += max(dp[temp][0],dp[temp][1]);//选择最大的存入 
        } 
    }
}

int main(){
    while(scanf("%d",&n) != EOF){
    //scanf("%d",&n);
    memset(fa,0,sizeof(fa));
    memset(dp,0,sizeof(dp));
    memset(vis,0,sizeof(vis));

    for(int i = 1;i<=n;i++) {
        scanf("%d",&dp[i][1]);
        son[i].clear(); 
    }

    int x,y;
    //memset(son,0,sizeof(son));

    int root = 0;
    while(scanf("%d%d",&x,&y) != EOF && x && y){
        son[y].push_back(x);//一个节点可以有好多的子节点 
        fa[x] = y;//一个节点只有一个父亲节点
        if(root == 0) root = y; 
    }
    while(fa[root]){
        root = fa[root];//找到根节点 
    }

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

}
    return 0;
}