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

题目链接

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

---

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:0f[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();      }}