题目描述
某大学有 n n n 个职员,编号为 1 … n 1\ldots n 1…n。
他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。
现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 r i r_i ri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。
所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入格式
输入的第一行是一个整数 n n n。
第 2 2 2 到第 ( n + 1 ) (n + 1) (n+1) 行,每行一个整数,第 ( i + 1 ) (i+1) (i+1) 行的整数表示 i i i 号职员的快乐指数 r i r_i ri。
第 ( n + 2 ) (n + 2) (n+2) 到第 2 n 2n 2n 行,每行输入一对整数 l , k l, k l,k,代表 k k k 是 l l l 的直接上司。
输出格式
输出一行一个整数代表最大的快乐指数。
样例 #1
样例输入 #1
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
样例输出 #1
5
提示
数据规模与约定
对于 100 % 100\% 100% 的数据,保证 1 ≤ n ≤ 6 × 1 0 3 1\leq n \leq 6 \times 10^3 1≤n≤6×103, − 128 ≤ r i ≤ 127 -128 \leq r_i\leq 127 −128≤ri≤127, 1 ≤ l , k ≤ n 1 \leq l, k \leq n 1≤l,k≤n,且给出的关系一定是一棵树。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=6e3+5;
int n,m,a[N],f[N],g[N],d[N]; //f选 g不选
vector<int>e[N];
void dfs(int u){
f[u]=a[u];
for(int i=0;i<e[u].size();i++){
int x=e[u][i];
dfs(x);
f[u]+=g[x]; //选择 u,则不能选择其子节点
g[u]+=max(f[x],g[x]); //不选 u,则可选也可不选
}
}
void solve(){
scanf("%lld",&n);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
}
for(int i=1;i<n;i++){
int x,y;
scanf("%lld%lld",&x,&y);
e[y].push_back(x);
d[x]++;
}
int root;
for(int i=1;i<=n;i++) if(d[i]==0){ //找出根节点
root=i;
break;
}
dfs(root);
int ans=0;
cout<<max(f[root],g[root]);
}
signed main(){
int t=1;
// scanf("%d",&t);
while(t--) solve();
return 0;
}