POJ - 3342 Party at Hali-Bula（树形dp）

题目链接：点击这里

题目大意：
$n$ 个人参加一场聚会，给出 $n$ 个人的工作关系树，一个人和他的顶头上司不能同时参加，求聚会达到的最大人数并判断邀请的最大人数名单是否唯一

题目分析：
对于最大人数我们和上司的舞会一样，用 $dp[i][0/1]$ 分别表示 $i$ 不去/去参加舞会，用树形 $dp$ 直接转移即可
对于最大人数情况是否唯一，我们可以再开一个数组 $f[i][0/1]$ 分别表示 $i$ 不去/去参加舞会时状态是否唯一，我们容易知道一个性质，如果其子节点中有一个方案数不唯一则其方案数不唯一，通过树形 $dp$ 来转移这个性质即可

具体细节见代码：

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define int  ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e5+5;
const int mod = 998244353;
const double pi = acos(-1);
const double eps = 1e-8;
int n,cnt,dp[maxn][2];
bool f[maxn][2];
vector<int>g[maxn];
map<string,int>mp;
string a,b;
void dfs(int u,int fa)
{
	dp[u][0] = 0,dp[u][1] = 1;
	f[u][0] = f[u][1] = true;
	for(int i = 0;i < g[u].size();i++)
	{
		int to = g[u][i];
		if(to == fa) continue;
		dfs(to,u);
		dp[u][0] += max(dp[to][1],dp[to][0]);
		dp[u][1] += dp[to][0];
		if(!f[to][0]) f[u][1] = false;
		if(dp[to][0] == dp[to][1])  f[u][0] = false;
		if(dp[to][0] > dp[to][1] && !f[to][0]) f[u][0] = false;
		if(dp[to][0] < dp[to][1] && !f[to][1]) f[u][0] = false;
	}
}
signed main()
{
	while(cin>>n && n)
	{
		mp.clear(); cnt = 0;
		for(int i = 1;i <= n;i++) g[i].clear();
		cin>>a;mp[a] = ++cnt;
		for(int i = 1;i < n;i++)
		{
			cin>>a>>b;
			if(mp.count(a) == 0) mp[a] = ++cnt;
			if(mp.count(b) == 0) mp[b] = ++cnt;
			g[mp[a]].push_back(mp[b]);
			g[mp[b]].push_back(mp[a]);
		}
		dfs(1,1);
		cout<<max(dp[1][0],dp[1][1])<<" ";
		if(dp[1][0] == dp[1][1]) puts("No");
		else if(dp[1][0] < dp[1][1]) puts(f[1][1] ? "Yes" : "No");
		else puts(f[1][0] ? "Yes" : "No");
	}
	return 0;
}