Uva-1220 Party at Hali-Bula

题目链接：Party at Hali-Bula

题目大意：给一棵树，父节点和其子结点不能同时染色，问最多可以选取多少个结点染色，如果解唯一则输出Yes否则输出No。

解题思路：树形DP，dp[i][0]表示标号为i的节点不染色的最大选取值，dp[i][1]则是染色时的情况。更新的时候再判断一下唯一性：

1.dp[v][0] == dp[v][1]子节点是否染色都一样，则dp[u][0]不唯一。
2.子节点必须(不)染色，且此自最优解不唯一，则dp[u][0]不唯一。

代码如下：

#include <map>#include <set>#include <cmath>#include <queue>#include <stack>#include <cstdio>#include <cstdlib>#include <cstring>#include <iomanip>#include <iostream>#include <algorithm>using namespace std;typedef long long ll;typedef pair<int, int> P;const int inf  = 0x3f3f3f3f;const int maxn = 2e4 + 15;bool Yuitsu, foo[205][2];int n, dp[205][2];vector<int> chd[205];map<string, int> ha;void DFS(int u){    if(dp[u][0] != -1 && dp[u][0] != -1) return;    if(!chd[u].size()){        dp[u][0] = 0; dp[u][1] = 1;         return;    }    dp[u][1] = 1; dp[u][0] = 0;    for(int i = 0; i < chd[u].size(); i++){        int v = chd[u][i];        DFS(v);         dp[u][1] += dp[v][0];          if(foo[v][0]) foo[u][1] = true;        dp[u][0] += max(dp[v][0], dp[v][1]);        if(dp[v][0] == dp[v][1]) foo[u][0] = true;        if(dp[v][0] > dp[v][1] && foo[v][0]) foo[u][0] = true;        if(dp[v][1] > dp[v][0] && foo[v][1]) foo[u][0] = true;    }    return;}int main(){#ifdef NEKO    freopen("Nya.txt", "r", stdin);#endif    ios::sync_with_stdio(false);  cin.tie(0);    string l, r;    while(cin >> n && n){        for(int i = 0; i <= n; i++) chd[i].clear();        memset(dp, -1, sizeof dp); ha.clear();        memset(foo, false, sizeof foo);        cin >> l; ha[l] = 0; Yuitsu = true;        for(int i = 1; i < n; i++){            cin >> l >> r;            if(!ha.count(l)) ha.insert({ l, ha.size() });            if(!ha.count(r)) ha.insert({ r, ha.size() });            chd[ha[r]].push_back(ha[l]);        }        DFS(0);        if(dp[0][1] == dp[0][0]) Yuitsu = false;        if(dp[0][1] > dp[0][0] && foo[0][1]) Yuitsu = false;        if(dp[0][1] < dp[0][0] && foo[0][0]) Yuitsu = false;        int ans = max(dp[0][1], dp[0][0]);        cout << ans << ' ' << (Yuitsu == true ? "Yes" : "No") << endl;    }    return 0;}