HDU 2412

HDU 1520的升级版而已~

#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
using namespace std;
const int maxn=205,inf=1<<30;
struct node
{
    int s,f;
    node(int s=0,int f=0):s(s),f(f){}
}dp[maxn][2];
int w[maxn],in[maxn];
int n,m;
vector<int>G[maxn];
void dfs(int u,int fa)
{

    dp[u][1].s=1;dp[u][1].f=0;
    dp[u][0].s=0;dp[u][0].f=0;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        dp[u][1].s+=dp[v][0].s,dp[u][1].f=dp[v][0].f;
        dp[u][0].s+=max(dp[v][1].s,dp[v][0].s);
        if(dp[v][1].s>dp[v][0].s&&dp[v][1].f) dp[u][0].f=1;
        else if(dp[v][1].s<dp[v][0].s&&dp[v][0].f) dp[u][0].f=1;
        else if(dp[v][1].s==dp[v][0].s) dp[u][0].f=1;
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    string s,a,b;
    while(cin>>n,n)
    {
        map<string,int>mp;
        int rt,cnt=1;
        for(int i=0;i<=n;i++) G[i].clear();
        cin>>s;mp[s]=cnt++;rt=cnt-1;
        for(int i=1;i<n;i++)
        {
            cin>>a>>b;
            if(mp[a]==0) mp[a]=cnt++;
            if(mp[b]==0) mp[b]=cnt++;
            G[mp[b]].push_back(mp[a]);
        }
        dfs(rt,-1);
        int s,f,v=rt;
        if(dp[v][1].s>dp[v][0].s) s=dp[v][1].s,f=dp[v][1].f;
        else if(dp[v][1].s<dp[v][0].s) s=dp[v][0].s,f=dp[v][0].f;
        else s=dp[v][1].s,f=1;
        if(f==0) cout<<s<<" Yes"<<endl;
        else cout<<s<<" No"<<endl;
    }
    return 0;
}