HDU 2412 Party at Hali-Bula(树形DP)

Party at Hali-Bula

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;
#define maxn 65540
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 300;

vector<int> child[N];//?号他的儿子，描述父子关系
map<string, int> v;//姓名、编号
int dp[N][2];
bool f[N][2]; //唯一性判断
void DFS(int u) {//u:人员的编号
    if(child[u].size() == 0) {//到了叶子节点了
        dp[u][0 ]= 0;
        dp[u][1] = 1;
        return;
    }

    int size = child[u].size();
    for(int i = 0; i < size; i++) {
        DFS(child[u][i]);//先把孩子撸了，再更新父亲

        //父亲来，孩子不来的情况
        dp[u][1] += dp[ child[u][i] ][0];//孩子不来,父亲来总人数
        if(f[ child[u][i] ][0])//孩子不来有多种
            f[u][1] = 1;//父亲来有多种

        //父亲不来，孩子可以来 也可以不来
        dp[u][0] += max(dp[ child[u][i] ][0], dp[ child[u][i] ][1]);
        if(dp[ child[u][i] ][0] > dp[ child[u][i] ][1]) {
            if(f[ child[u][i] ][0])//孩子不来有多种
                f[u][0] = 1;//父亲不来有多种
        } else if(dp[ child[u][i] ][1] == dp[ child[u][i] ][0]) {
                f[u][0] = 1;//父亲不来有多种
        } else if(dp[ child[u][i] ][0] < dp[ child[u][i] ][1]) {
            if(f[ child[u][i] ][1])//孩子来有多种
                f[u][0] = 1;//父亲不来有多种
        }
    }

    dp[u][1]++;//不要忘记加上父亲自己
}

int main() {
    freopen("data.in", "r", stdin);
    int n;

    while(~scanf("%d", &n) && n) {
        memset(dp, 0, sizeof(dp));
        memset(f, 0, sizeof(f));
        int top = 1;
        string s1, s2;
        cin >> s1;//big boss
        v[s1] = top++;
        for(int i = 1; i < n; i++) {
            cin >> s1 >> s2;
            if(!v[s1])
                v[s1] = top++;
            if(!v[s2])
                v[s2] = top++;
            child[v[s2]].push_back(v[s1]);
        }

        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], 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++)
            child[i].clear();

        v.clear();
    }

    return 0;
}