AcWing
285. 没有上司的舞会
树形DP. 开dp数组f[N][2].
f[i][0]：以i为根的子树，不选择i这个根节点的所有方案的最大价值；
f[i][1]：以i为根的子树，选择i这个根节点的所有方案的最大价值。
状态转移：
- 对于
f[i][0]，即不选择i这个根节点，那么其所有的子树可以选择根节点，也可以不选择根节点，选取这两个中价值最大的一种，对所有子树的价值加和。f[i][0] = sum_j{max{f[j][0], f[j][1]}}; - 对于
f[i][1]，即选择i这个根节点，那么其所有的子树都不能选择根节点，对所有子树的价值加和，再加上根节点i本身的价值。f[i][1] = w[i] + sum_j{f[j][0]}.
通过dfs，从叶子节点递归求取f[i][0]、f[i][1].
题目未给出根节点，需要自行判断。
建树可以采用一般的建图方式，即邻接表，注意初始化表头为-1.
#include <bits/stdc++.h>
using namespace std;
int n;
const int N = 6010;
int w[N];
int h[N], e[N], ne[N], idx;
bool hasfather[N];
int f[N][2];
void add(int a, int b){
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u){
f[u][1] = w[u];
for(int i = h[u]; ~i; i = ne[i]){
int v = e[i];
dfs(v);
f[u][0] += max(f[v][0], f[v][1]);
f[u][1] += f[v][0];
}
}
int main(){
cin >> n;
memset(h, -1, sizeof h);
for(int i = 1; i <= n; i++) cin >> w[i];
for(int i = 1; i <= n - 1; i++){
int a, b;
cin >> a >> b;
hasfather[a] = true;
add(b, a);
}
int root;
for(int i = 1; i <= n; i++) if(!hasfather[i]) root = i;
dfs(root);
cout << max(f[root][0], f[root][1]) << endl;
return 0;
}