POJ 3342 Party at Hali-Bula 树形DP 最大独立集

题意：一个公司想举办聚会，但是为了气氛，如果邀请一个员工的话，是不会再邀请他的直接上司。这个公司的人员结构是树形的，每个人至多有一名直接上司。给出这个公司的人员结构，求出能参加聚会的人数的最大值，同时判断方案是否唯一。
思路：同样的树的最大独立集，不同的是还需要判断方案是否唯一。其实，如果我们求出了方案数，就可以知道方案是否是唯一的。
题目中是给出的每个人的名字，直接用map离散一下就可以得到数字了。
可以注意到，节点u和是否被选就能描述对应的状态。
所以定义dp[u][0]为不选节点u，以节点u为根的子树得到的最大独立集，f[u][1]为对应的方案数。
dp[u][1]为选节点u，以节点u为根的子树得到的最大独立集，f[u][0]为对应的方案数。
下面考虑状态转移：
1.因为dp[u][1]为选择节点u，那就不能选他的儿子节点，只能选择dp[v][0]
所以：

$$dp[u][1] = 1 +\sum_{v \in son(u)}dp[v][0]， f[u][1] = \prod_{v \in son(u)}f[v][0]$$
2.dp[u][0]表示不选择节点u，那对于其儿子节点，我们可以选也可以不选
所以：
$$dp[u][0] =\sum_{v \in son(u)}max(dp[v][0],dp[v][1])$$
方案计数为：
$$f[u][0] = \prod_{v \in son(u)}{\begin{cases} f[v][0], & \text{$dp[v][0] \gt dp[v][1] $} \\ f[v][0] + f[v][1], & \text{$dp[v][0] = dp[v][1] $}\\ f[v][1], & \text{$dp[v][0] \lt dp[v][1]$} \end{cases} }$$
初始条件：
$$dp[u][1] = 1, dp[u][0] = 0, f[u][1] = f[u][0] = 1$$
最终答案，当dp[1][0] = dp[1][1] 或 取到最大值的方案数不唯一， 最终方案不唯一。
代码如下：

#include <iostream>
#include <cstring>
#include <map>
#include <string>
#include <algorithm>

using namespace std;

const int MAX = 20000;

int N;
int to[MAX*2],nxt[MAX<<1],head[MAX],tot;
int dp[MAX][2],f[MAX][2];
map<string,int> ma;


void init()
{
    memset(head,-1,sizeof(head));
    tot = 0;
    ma.clear();
}

void addedge(int u, int v)
{
    to[tot] = v, nxt[tot] = head[u];
    head[u] = tot++;
    to[tot] = u, nxt[tot] = head[v];
    head[v] = tot++;
}

int insert(string s)
{
    if(ma[s] == 0)
        return ma[s] = ma.size();
    else
        return ma[s];
}

void dfs(int u, int p)
{
    dp[u][1] = 1,dp[u][0] = 0;
    f[u][1] = f[u][0] = 1;
    for(int i = head[u]; ~i; i = nxt[i]){
        int v = to[i];
        if(v == p) continue;
        dfs(v,u);

        dp[u][1] += dp[v][0];
        f[u][1] *= f[v][0];

        if(dp[v][0] > dp[v][1]){
            dp[u][0] += dp[v][0];
            f[u][0] *= f[v][0];
        }
        else if(dp[v][0] < dp[v][1]){
            dp[u][0] += dp[v][1];
            f[u][0] *= f[v][1];
        }
        else{
            dp[u][0] += dp[v][1];
            f[u][0] *= (f[v][1] + f[v][0]);
        }
    }
}


int main(void)
{
    //freopen("input.txt","r",stdin);
    while(cin>>N,N){
        init();
        string s1,s2;
        cin>>s1;
        int root = insert(s1);
        for(int i = 0; i < N - 1; ++i){
            cin>>s1>>s2;
            int u = insert(s1),v = insert(s2);
            addedge(u,v);
        }
        dfs(root,0);
        int ans = max(dp[root][0],dp[root][1]);
        cout << ans <<' ';
        if(dp[root][0] == dp[root][1] || (ans == dp[root][0] && f[root][0] != 1)
           || (ans == dp[root][1] && f[root][1] != 1))
            cout<<"No"<<endl;
        else
            cout <<"Yes"<<endl;;
    }
    return 0;
}