Party at Hali-Bula HDU - 2412 （树形DP判断是否唯一最优解）

Party at Hali-Bula

 题目链接：HDU - 2412 

题意：笔者要开一个party，邀请公司的所有同事，当然了包括各位boss，但是每个人都不会和他的直接上司同事参加party，问可最多可以有多少人来参加party，并判断是否只有唯一解；

第一问很简单，就是树形dp入门了，每个人只有两种情况，1参加，0不参加；参加，则他的手下一定不参加；不参加，则他的手下可能参加，也可能不参加；状态转移方程为：

dp[i][0]=max(dp[son][1], dp[son][0]);

dp[i][1]=dp[son]]0]+1;

现在重点来剖析一下第二问；现在假设flag[i][1]=1表示i参加party是唯一解，=0不是；flag[i][0]=1表示i不参加party是唯一解，=0不是；问题是如何判断；

首先对于所有叶子节点来说，flag[i][1]=flag[i][0]=1;

对于非叶节点i；  son是i的孩子节点：

当假设i不参加party时：

如果dp[son][1]>dp[son][0]说明son要参加，如果flag[son][1]=0那么flag[i][0]必定是0；

如果dp[son][0]>dp[son][1]说明son不参加，所以若flag[son][0]=0那么flag[i][0]必为0；

如果dp[son][0]=dp[son][1]说明son参不参加对dp[i][0]没有影响，此时flag[i][0]为0；

当假设i参加party时：

此状态是由dp[son][0]推过来的，那么只要flag[son][0]=0,则flag[i][1]=0;

#include <bits/stdc++.h>
using namespace std;
map<string, int>name;
vector<int> vec[210];
bool flag[210][3];
int dp[210][3];
void dfs(int root){
	dp[root][0]=0;
	dp[root][1]=1;
	flag[root][0]=flag[root][1]=true;
	for(int i=0; i<vec[root].size(); i++){
		int son=vec[root][i];
		dfs(son);
		dp[root][0]+=max(dp[son][1], dp[son][0]);
		dp[root][1]+=dp[son][0];
		if(dp[son][0]>dp[son][1]&&flag[son][0]==0) flag[root][0]=0;
		if(dp[son][1]>dp[son][0]&&flag[son][1]==0) flag[root][0]=0;
		if(dp[son][0]==dp[son][1]) flag[root][0]=0;
		if(flag[son][0]==0) flag[root][1]=0;
	}
}
int main(){
	int n;
	while(scanf("%d", &n), n){
		int cnt=0;
		for(int i=0; i<=n; i++) vec[i].clear();
		name.clear();
		string a, b;
		cin >> a;
		name[a]=++cnt;
		for(int i=1; i<n; i++){
			cin >> a >> b;
			if(!name.count(a)) name[a]=++cnt;
			if(!name.count(b)) name[b]=++cnt;
			vec[name[b]].push_back(name[a]);
		}
		dfs(1);
		printf("%d ", max(dp[1][0], dp[1][1]));
		if(dp[1][0]>dp[1][1]){
			if(flag[1][0]) printf("Yes\n");
			else printf("No\n");
		}
		else if(dp[1][1]>dp[1][0]){
			if(flag[1][1]) printf("Yes\n");
			else printf("No\n");
		}
		else printf("No\n");
	}
	return 0;
}