Uva-1220-Party at Hali-Bula

继续简单的树形DP

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#include<string>
#include<vector>
using namespace std;
const int maxn=210;
int n,cnt,dp[maxn][2];
bool unique[maxn][2];
char name[110];
vector<int> g[maxn];
map<string,int> s;
void Init()
{
    cnt=0;
    memset(dp,0,sizeof(dp));
    memset(unique,0,sizeof(unique));
    s.clear();
    for(int i=1;i<=n;i++)
        g[i].clear();
}
void DFS(int u)
{
    if(g[u].size()==0)
    {
        dp[u][0]=0;
        dp[u][1]=1;
        return;
    }
    int size=g[u].size();
    for(int i=0;i<size;i++)
    {
        DFS(g[u][i]);
        if(unique[g[u][i]][0])
            unique[u][1]=1;
        dp[u][1]+=dp[g[u][i]][0];
        if(dp[g[u][i]][0]>dp[g[u][i]][1])
        {
            dp[u][0]+=dp[g[u][i]][0];
            if(unique[g[u][i]][0])
                unique[u][0]=1;
        }
        else
        {
            dp[u][0]+=dp[g[u][i]][1];
            if(dp[g[u][i]][1]==dp[g[u][i]][0]||unique[g[u][i]][1])
                unique[u][0]=1;
        }
    }
    dp[u][1]++;
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        Init();
        scanf("%s",name);
        s[string(name)]=++cnt;
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%s",name);
            u=s[string(name)];
            if(!u)
            {
                s[string(name)]=++cnt;
                u=cnt;
            }
            scanf("%s",name);
            v=s[string(name)];
            if(!v)
            {
                s[string(name)]=++cnt;
                v=cnt;
            }
            g[v].push_back(u);
        }
        DFS(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],unique[1][1]?"No":"Yes");
        else
            printf("%d %s\n",dp[1][0],unique[1][0]?"No":"Yes");
    }
    return 0;
}