UVA 1220 树形dp

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 2E2 + 10;
vector<int>v[maxn];
map<string, int>mp;
int dp[maxn][3], n, tol;
bool vis[maxn][3];
void Treedp(int u)
{
	if (v[u].size() == 0) {dp[u][0] = 0; dp[u][1] = 1; return;}
	int size = v[u].size();
	for (int i = 0; i < size; i++)
	{
		int t = v[u][i]; Treedp(v[u][i]);
		if (vis[t][0]) vis[u][1] = 1;
		dp[u][1] += dp[t][0];
		dp[u][0] += dp[t][(dp[t][0] > dp[t][1] ? 0 : 1)];
		if (dp[t][0] > dp[t][1] && vis[t][0] || dp[t][0] < dp[t][1] && vis[t][1] || dp[t][1] == dp[t][0])
			vis[u][0] = 1;
	}
	dp[u][1]++;
}
int main(int argc, char const *argv[])
{
	while (~scanf("%d", &n) && n)
	{
		string s1, s2; tol = 0;
		memset(dp, 0, sizeof(dp));
		memset(vis, 0, sizeof(vis));
		cin >> s1; mp[s1] = ++tol;
		for (int i = 1; i < n; i++)
		{
			cin >> s1 >> s2;
			if (!mp[s1]) mp[s1] = ++tol;
			if (!mp[s2]) mp[s2] = ++tol;
			v[mp[s2]].push_back(mp[s1]);
		}
		Treedp(1);
		if (dp[1][1] == dp[1][0]) printf("%d No\n", dp[1][1]);
		else if (dp[1][1] > dp[1][0]) printf("%d %s\n", dp[1][1], vis[1][1] ? "No" : "Yes");
		else printf("%d %s\n", dp[1][0], vis[1][0] ? "No" : "Yes");
		for (int i = 1; i <= n; i++) v[i].clear();
		mp.clear();
	}
	return 0;
}

有一个公司员工要举行聚会，要求任意一个人不能和他的直接上司同时到场，一个员工只有一个支系上司，现在求最多有多少人到场，并且方案是否唯一

转化为求树的最大独立集问题。用树形dp解决。

定义dp[x][0]：表示在 i 点不选 i 点的以 x 为子树的最大独立集 而dp[x][1] 表示x到场的最大独立集
定义f [x][0]：表示以x为根且x点不选的子树是否唯一 ，f[x][1]表示以x为根且x选的子树是否唯一
状态转移方程：dp [ x ] [ 1 ] + = dp [ child ] [ 0 ] ;
 dp [ x ] [ 0 ] +  = max ( dp [ child ] [ 0 ] , dp [ child ] [ 1 ] );