紫书动规 例题9-13 UVA - 1220 Party at Hali-Bula 树形dp

题目链接:

https://vjudge.net/problem/UVA-1220

题意:

题解:

树形dp： 树的最大独立集问题

d[u][0]:=不选u能得到的最大人数 d[u][0]=sum{max(d[v][0],d[v][1]}

d[u][1]:=选u能得到的最大人数 d[u][1]=sum{d[v][0]}

判断唯一性就是看当前点的孩子是不是不唯一的。初始唯一

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e5+10;

int n,cnt;
map<string,int> mp;
string s1,s2;
int d[300][2],f[300][2];

vector<int> g[300];

int ID(string s){
if(!mp.count(s)) mp[s] = cnt++;
return mp[s];
}

int dp(int u,int k){
d[u][k] = k;
f[u][k] = 1;
for(int i=0; i<(int)g[u].size(); i++){
int v = g[u][i];
if(k == 1){
d[u][k] += dp(v,0);
if(!f[v][0]) f[u][k]=0;
}else{
d[u][k] += max(dp(v,0),dp(v,1));
if(d[v][0]==d[v][1]) f[u][k]=0;
else if(d[v][0]>d[v][1] && !f[v][0]) f[u][k]=0;
else if(d[v][1]>d[v][0] && !f[v][1]) f[u][k]=0;
}
}
return d[u][k];
}

int main(){
while(scanf("%d",&n) && n){
cnt = 0;
for(int i=0; i<=n; i++) g[i].clear();
mp.clear();

cin >> s1;
ID(s1);
for(int i=1; i<n; i++){
cin >> s1 >> s2;
g[ID(s2)].push_back(ID(s1));
}

int ans = max(dp(0,0),dp(0,1));
cout << ans << " ";

bool unique = true;
if(d[0][0] == d[0][1]) unique=false;
if(d[0][0]>d[0][1] && !f[0][0]) unique=false;
if(d[0][1]>d[0][0] && !f[0][1]) unique=false;

if(unique) puts("Yes");
else puts("No");
}

return 0;
}