大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。
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;
}
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/168855.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...