Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺
Ural 大学有 N 名职员，编号为 1∼N。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。
现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。
输入格式
第一行一个整数 N。
接下来 N 行，第 i 行表示 i 号职员的快乐指数 Hi。
接下来 N−1 行，每行输入一对整数 L,K，表示 K 是 L 的直接上司。
输出格式
输出最大的快乐指数。
数据范围
1≤N≤6000,
−128≤Hi≤127
输入样例：
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例：
5
题解
f[i][0]：节点0没有选，树的最大值
f[i][1]：节点0选了，树的最大值
#include<bits/stdc++.h>
using namespace std;
const int N = 6e3 + 10;
int f[N][2],a[N],in[N];
struct Edge{
int v,next;
}edge[N];
int head[N],cnt;
void add(int u,int v){
edge[cnt].v = v;
edge[cnt].next = head[u];
head[u] = cnt++;
}
int dfs(int u){
for(int i = head[u];~i;i = edge[i].next){
int v = edge[i].v;
dfs(v);
f[u][0] += max(f[v][1],f[v][0]);
f[u][1] += f[v][0];
}
f[u][1] += a[u];
}
int main(){
int n,root = -1;
cin>>n;
memset(head,-1,sizeof head);
int x,y;
for(int i = 1;i <= n;i ++)cin>>a[i];
for(int i = 0;i < n - 1;i ++){
cin>>x>>y;
add(y,x);
in[x] ++;
}
for(int i = 1;i <= n;i ++)
if(in[i] == 0)root = i;
dfs(root);
cout<<max(f[root][0],f[root][1]);
return 0;
}
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/168674.html原文链接：https://javaforall.net
