为了庆祝乌拉尔州立大学建校80周年,将举行一个晚会。这所大学有雇员的等级结构。这意味着主管关系形成了一棵树,根植于校长V.E.Tretyakov。为了让每个人都觉得聚会有趣,校长不希望员工和他或她的直接上司都在场。人事处对每个员工的好客性进行了评估,所以每个人都有自己的号码(等级)。你的任务是列出一份客人的名单,尽可能多地列出客人的欢乐程度。
输入
雇员的编号从1到N。输入的第一行包含一个数字N.1<=N<=6000。随后的N行中的每一行都包含相应员工的欢乐度评级。欢乐等级是一个从-128到127的整数。之后是N–1行,用于描述主管关系树。树规范的每一行都有以下形式:
好的
这意味着第K个员工是第L个员工的直接主管。输入以行结束
0 0
输出
输出应包含客人评分的最大总和。
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
5
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<queue>
#include<stack>
#include<string.h>
using namespace std;
vector<int>G[10000];
const int p=100000;
int dp[p][2];
int fa[p];
void f(int u)
{
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i]; //遍历所有人
f(v);
}
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
dp[u][0] = dp[u][0] + max(dp[v][1],dp[v][0]); //进行选择判断是否参加
dp[u][1] = dp[u][1] + dp[v][0];
}
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
scanf("%d",&dp[i][1]);
dp[i][0]=0;
fa[i]=-1;
}
int u,v; //u 员工 v老板
while(cin>>u>>v)
{
if(u == 0&& v == 0)
break;
G[v].push_back(u);
fa[u] = v;
}
int root=1;
while(fa[root]!=-1)
root=fa[root]; //找到最终老板
f(root);
printf("%d\n",max(dp[root][1],dp[root][0]));
return 0;
}