当前位置: 代码迷 >> 综合 >> UVA1220[Party at Hali-Bula] 树上动态规划
  详细解决方案

UVA1220[Party at Hali-Bula] 树上动态规划

热度:86   发布时间:2023-11-06 08:16:56.0

题目链接

题意:一个公司员工要举行聚会,要求任意一个人不能和他的直接上司同时到场,一个员工只有一个支系上司,现在求最多有多少人到场,并且方案是否唯一。


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();  }
}