AcWing 285. 没有上司的舞会--树形DP

AcWing 285. 没有上司的舞会

题目描述

Ural 大学有 N 名职员，编号为 1∼N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式

第一行一个整数 N。

接下来 N 行，第 i 行表示 i 号职员的快乐指数 Hi。

接下来 N−1 行，每行输入一对整数 L,K，表示 K 是 L 的直接上司。

输出格式

输出最大的快乐指数。

数据范围

$1\le N\le 6\ast 10^3,-128\le Hi\le 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;
}