题目描述
Ural 大学有 N 名职员，编号为 1∼N。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。
现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。
输入格式
第一行一个整数 N。
接下来 N 行，第 i 行表示 i 号职员的快乐指数 Hi。
接下来 N−1 行，每行输入一对整数 L,K，表示 K 是 L 的直接上司。
输出格式
输出最大的快乐指数。
数据范围
1 ≤ N ≤ 6 ∗ 1 0 3 , − 128 ≤ H i ≤ 127 1≤N≤6*10^3, −128≤Hi≤127 1≤N≤6∗103,−128≤Hi≤127
输入样例：
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例：
5
#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template<class...Args>
void debug(Args... args) {//Parameter pack
auto tmp = { (cout << args << ' ', 0)... };
cout << "\n";
}
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 1e5 + 5;
const ll INF = 0x7fffffff;
const ll MOD = 1e9 + 7;
int _head[N], _next[N], val[N],idx;
void init() {
memset(_head, -1, sizeof _head);
}
void add(int u, int v) {
val[idx] = v, _next[idx] = _head[u], _head[u] = idx++;
}
int f[N][2];
bool has_fa[N];
int happy[N];
void dfs(int u) {
f[u][1] = happy[u];
for (int i = _head[u]; ~i; i = _next[i]) {
int v = val[i];
dfs(v);
f[u][1] += f[v][0];
f[u][0] += max(f[v][0], f[v][1]);
}
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int n;
cin >> n;
for (int i = 1; i <= n; i++)cin >> happy[i];
init();
for (int i = 0; i < n - 1; i++) {
int u, v;
cin >> u >> v;
add(v, u);
has_fa[u] = true;
}
int root = 1;
while (has_fa[root])root++;
dfs(root);
cout << max(f[root][0], f[root][1]);
return 0;
}