题目链接
题意:一个公司员工要举行聚会,要求任意一个人不能和他的直接上司同时到场,一个员工只有一个支系上司,现在求最多有多少人到场,并且方案是否唯一。
solution:用dp[u][0], dp[u][1] 分别代表 u结点 不选 或 选, f[u][0], f[u][1], 代表对应的方案是否唯一。
dp[u][0]+=max(dp[v][0],dp[v][1]) v是u的字结点
dp[u][1]+=dp[v][0]
f[u][1]=(f[v][0]==1)?1:0
f[u][0]与dp[u][0]类似
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int N = 205;
vector<int> G[N];
map<string,int> mp;
int dp[N][3], f[N][3], tot;void d(int u){if ( !G[u].size() ){dp[u][0]=0;dp[u][1]=1;return ; }int size=G[u].size();for ( int i=0; i<size; i++ ){int v=G[u][i];d(v);dp[u][1]+=dp[v][0];if( f[v][0] ) f[u][1]=1;if( dp[v][1]>dp[v][0] ){dp[u][0]+=dp[v][1];if( f[v][1] ) f[u][0]=1;}else {dp[u][0]+=dp[v][0];if( dp[v][1]==dp[v][0] || f[v][0] ) f[u][0]=1;} }dp[u][1]++;
}
int id(string str){if ( mp.count(str) ) return mp[str];mp[str]=++tot;return tot;
}
int main(){int n;while( scanf("%d", &n )==1 && n ){memset(dp,0,sizeof(dp));memset(f,0,sizeof(f));tot=0;string s1, s2;cin>>s1;int root=id(s1);for ( int i=1; i< n; i++ ){cin>>s1>>s2;G[id(s2)].push_back(id(s1));}d(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],f[1][1]?"No":"Yes"); else printf("%d %s\n",dp[1][0],f[1][0]?"No":"Yes"); for(int i = 1;i<=n;i++) G[i].clear(); mp.clear(); }
}