题意:
公司参加聚会,要求员工不能和他的上司同时参加,求最多能参加几个人并且判断解是否唯一。
要点:
树型DP的经典题,用dp[u][1]表示选取u的最大值,dp[u][0]表示不选取u的最大值,容易得到状态转移方程为dp[u][1] += dp[v][0]和dp[u][0] += max(dp[v][1], dp[v][0]),可以看到如果dp[u][0]>dp[u][1]时,dp[v][1]==dp[v][0]会造成多解,那么我们最后遍历一遍即可判断是否有多解,还有一种情况是dp[0][1]==dp[0][0],这当然也有多解。这题用了map,稍微学了一下,还是很好用的。
15573797 | Seasonal | 3342 | Accepted | 212K | 16MS | C++ | 1336B | 2016-05-31 09:15:14 |
#include<cstdio>
#include<cstring>
#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
vector<int> edge[205];
int dp[205][2];
void dfs(int u,int p)
{
int i, v;
dp[u][0] = 0;//不取当前值
dp[u][1] = 1;//取当前值
for (i = 0; i < edge[u].size(); i++)
{
int v = edge[u][i];
dfs(v, u);
dp[u][1] += dp[v][0];
dp[u][0] += max(dp[v][1], dp[v][0]);//这里状态转移会造成有不唯一的解
}
}
int main()
{
int n,i,j;
char boss[200],a[200],b[200];
map<string, int> mm;
while (scanf("%d", &n), n)
{
mm.clear();
for (i = 0; i < 205; i++)
edge[i].clear();
int tot = 0;
scanf("%s", boss);
mm[boss] = tot++;
for (i = 0; i < n - 1; i++)
{
scanf("%s%s", a, b);
if (mm.find(a) == mm.end()) mm[a] = tot++;
if (mm.find(b) == mm.end()) mm[b] = tot++;//如果map中没有b,就把b加入,对应的映射是tot
edge[mm[b]].push_back(mm[a]);
}
dfs(0, 0);
bool flag = true;
for (i = 0; i < n; i++)
{
flag = true;
if (dp[i][0]>dp[i][1])//只有取dp[i][0]时可能有多解
{
for (j = 0; j < edge[i].size(); j++)
{
if (dp[edge[i][j]][0] == dp[edge[i][j]][1])
{
flag = false;
break;
}
}
}
if (!flag)
break;
}
printf("%d ", max(dp[0][0], dp[0][1]));
if (!flag||dp[0][1]==dp[0][0])//还有一种是总体考虑boss要不要选造成的多解
printf("No\n");
else
printf("Yes\n");
}
return 0;
}