闫学灿acwing_二叉树外部路径内部路径

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新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;
}