补上题代码 hdu1520

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
 
using namespace std;
 
#define maxn 6005
struct node{
   int to,net;
}que[maxn<<];
int head[maxn];
int n;
int dp[maxn][],father[maxn];//dp[i][0]0表示不去，dp[i][1]1表示去了
bool visited[maxn];
int tot=;
 
void addedge(int u,int v){
    que[tot].to=v;
    que[tot].net=head[u];
    head[u]=tot++;
 
    que[tot].to=u;
    que[tot].net=head[v];
    head[v]=tot++;
 
}
 
void tree_dp(int node)
{
    int i;
    visited[node] = ;
    for(i=head[node]; i!=-; i=que[i].net)
    {
        int v=que[i].to;
        if(!visited[v]&&father[v] == node)//i为下属
        {
            tree_dp(v);//递归调用孩子结点，从叶子结点开始dp
            //关键
        )   dp[node][] += dp[v][];//上司来,下属不来
            dp[node][] +=max(dp[v][],dp[v][]);//上司不来，下属来、不来
        }
    }
}
 
int main()
{
    int i;
    int f,c,root;
    while(scanf("%d",&n)!=EOF)
    {
        tot=;
        memset(head,-,sizeof(head));
        memset(dp,,sizeof(dp));
        memset(father,,sizeof(father));
        memset(visited,,sizeof(visited));
        for(i=; i<=n; i++)
        {
            scanf("%d",&dp[i][]);
        }
        root = ;//记录父结点
        bool beg = ;
        while (scanf("%d %d",&c,&f),c||f)
        {
            addedge(c,f);
            father[c] = f;
            if( root == c || beg )
            {
                root = f;
            }
        }
        while(father[root])//查找父结点
            root=father[root];
        tree_dp(root);
        int imax=max(dp[root][],dp[root][]);
        printf("%d\n",imax);
    }
    return ;
 
}