POJ 2342 Anniversary party （树形dp）

题意：某大学有n个职员，编号为1~n。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

题解：树形dp
设$dp[x][0]$表示以x为根的子树，且x不参加舞会的最大快乐值；$dp[x][1]$表示以x为根的子树，且x参加了舞会的最大快乐值。则转移方程为：$dp[x][0]+=max(dp[y][0],dp[y][1])，dp[x][1]+=dp[y][0]+h[x]$，其中y是x的下属。

以root为根节点，递归子结点即可，最终答案为$max(dp[root][0],dp[root][1])$。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
#define maxn 6666
int h[maxn], f[maxn], dp[maxn][2], n, x, y, root;
vector<int> son[maxn];
void dfs(int x) {
	dp[x][0] = 0;
	dp[x][1] = h[x];
	int len = son[x].size();
	for (int i = 0; i < len; i++) {
		int y = son[x][i];
		dfs(y);
		dp[x][0] += max(dp[y][0], dp[y][1]);
		dp[x][1] += dp[y][0];
	}
}
int main() {
	while (~scanf("%d", &n)) {
		memset(dp, 0, sizeof(dp));
		memset(son, 0, sizeof(son));
		for (int i = 1; i <= n; i++) {
			scanf("%d", &h[i]);
		}
		while (scanf("%d%d", &x, &y) && x && y) {
			son[y].push_back(x);
			f[x] = y;
		}
		root = 1;
		while (f[root]) {
			root = f[root];
		}
		dfs(root);
		printf("%d\n", max(dp[root][0], dp[root][1]));
	}
	return 0;
}