【刷题】动态规划——树形DP：没有上司的舞会

题目链接

经典树形DP问题，父亲和孩子二者只能选1个。
也就是1条边至多选1个点，点与点之间没有没有联系，这样所选的点的集合也叫独立集。
独立集

求具有最大权值的独立集。

状态主要有两种：
1、f[u, 0]：从所有以u为根的子树中选，不选u这个点的方案
2、f[u, 1]：从所有以u为根的子树中选，选u这个点的方案

设s1、s2、…、sn是u的孩子

1、不选u，则可以在u的孩子中选，当然也可以挑不选孩子的方案，求最大即可。
f[u, 0] = max(f[s1, 0], f[s1, 1]) + max(f[s2, 0], f[s2, 1]) + ... + max(f[sn, 0], f[sn, 1])
2、选了u，那么只能挑不选孩子的方案，也就是f[s, 0]
f[u, 1] = f[s1, 0] + f[s2, 0] + ... + f[sn, 0]

最后的答案就是max(f[root, 0], f[root, 1])，root为根节点

用邻接链表来存储树，每个父亲存储它的孩子有哪些，将孩子用链表串起来。
head[i]存编号为i的节点在链表数组的位置，不妨设位置为j，val[j]存节点i的孩子节点编号s1，nxt[j]存节点i的下一个孩子s2在链表数组的位置（也就是head[s2]）。
令j = nxt[j]，直到j = -1为止，可以获得i的所有孩子的编号值val[j]

#include <iostream>
#include <cstring>
using namespace std;
const int N = 6005;
int n, h[N], f[N][2];
bool has_father[N];
int head[N], val[N], nxt[N], len;

void add_node(int a, int b) {
	val[len] = b;
	nxt[len] = head[a];
	head[a] = len ++;
}

void dfs(int u) {
	f[u][1] = h[u];
	for (int i = head[u]; i != -1; i = nxt[i]) {
		int s = val[i];
		dfs(s);
		
		f[u][0] += max(f[s][0], f[s][1]);
		f[u][1] += f[s][0];
	}
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i ++) {
		scanf("%d", &h[i]);
	}
	memset(head, -1, sizeof(head));
	int a, b;
	for (int i = 1; i <= n - 1; i ++) {
		scanf("%d%d", &a, &b);
		has_father[a] = true;
		add_node(b, a);
	}
	
	int root;
	for (int i = 1; i <= n; i ++) {
		if (!has_father[i]) {
			root = i;
			break;
		}
	}
	dfs(root);
	printf("%d\n", max(f[root][0], f[root][1]));
	return 0;
}