【UVA1220】Party at Hali-Bula

来源:互联网 发布:神仙道心动网络 编辑:程序博客网 时间:2024/05/29 14:19

要原题点这里题目。

题目大意:公司里n个人构成了一个树形网络,除了老板之外每个人都有一个直属上司,问在不能同时选一个人的直属上司的同时最多能选多少人。并判断人数最多时方案是否唯一。

题解

树形dp经典题。
这题的第一问就是树上的最大独立集的问题,第二问就是在dfs的基础上加一些判断就行了.
f[i][0]表示选当前这个人,f[i][1]表示不选当前这个人。所以f[i][0]+=f[v][1],f[i][1]+=max(f[v][1],f[v][0]);这题没要求全部覆盖所有点,所有不用记录父节点选的情况。现在讨论什么时候方案数不唯一。g[v][0/1]表示方案数是否唯一,如果f[v][0]==f[v][1]那么不唯一,因为选哪个都行。如果max取到的g[v][0]==0,方案数也不唯一,(如f[v][0]>f[v][1]&&g[v][0]==0).
#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<map>using namespace std;const int N=100010;map<string,int>mp;int f[N][3];bool g[N][3];int pre[N*2],last[N],other[N*2],num;int cnt;string s,ss;int getid(string x){if(!mp.count(x)) mp[x]=++cnt;return mp[x];}inline void add(int x,int y){num++;pre[num]=last[x];last[x]=num;other[num]=y;}void dfs(int x){f[x][1]=1;f[x][0]=0;g[x][1]=g[x][0]=1;for(int i=last[x];i;i=pre[i]){int v=other[i];dfs(v); f[x][1]+=f[v][0];f[x][0]+=max(f[v][0],f[v][1]);if(f[v][0]==f[v][1]) g[x][0]=0;else if(f[v][0]>f[v][1]&&g[v][0]==0) g[x][0]=0;else if(f[v][0]<f[v][1]&&g[v][1]==0) g[x][0]=0;if(g[v][0]==0) g[x][1]=0;}}int main(){int n;while(scanf("%d",&n)&&n){mp.clear();cnt=0;memset(last,0,sizeof(last));num=0;memset(f,0,sizeof(f));memset(g,0,sizeof(g));cin>>s; getid(s);for(int i=1;i<n;i++){cin>>s>>ss;int x=getid(s),y=getid(ss);add(y,x);}dfs(1);if(f[1][0]>f[1][1]&&g[1][0]) printf("%d Yes\n",f[1][0]);else if(f[1][0]<f[1][1]&&g[1][1]) printf("%d Yes\n",f[1][1]);else printf("%d No\n",max(f[1][1],f[1][0]));}return 0;}