POJ3342树状动态规划

题意：

一个公司有n个员工，但是除了第一个人是
大老板没有上司外，其他的人都有唯一的
一个上司，现在有个人要开party，但是对
邀请的人如果有他的直接上司，他就不高
兴，要求一个最大的高兴人数。

分析：

这就是树的最大独立集，但是对于判断唯一
对点u,dpu[u][0]表示u在独立集中时情况是
否唯一，dpu[u][1]表示u不在独立集中时情
况是否唯一。对叶子节点dpu[u][0]=dpu[u][0]=true
对非叶子节点，如果存在v，pre[v] = u,
dpu[v][1]=false,则dpu[u][0]=false
如果存在dp[v][1]>dp[v][0]且dpu[v][1]=false
则dpu[u][1]=false
如果存在dp[v][0]>dp[v][1]且dpu[v][0]=false
则dpu[u][1]=false
如果dp[v][0]=dp[v][1]则dpu[u][1]=false

代码：

#include <iostream>
#include <string>
#include <map>
#include <memory.h>
#include <fstream>
using namespace std;
 
int n,dp[210][2],tree[210][210];
string str1,str2;
 
void fun(int t)
{
    dp[t][0] = 0;dp[t][1] = 1;
    for(int i = 1;i <= n;i ++)
    if(tree[t][i])
    {
        fun(i);
        dp[t][0] += max(dp[i][0],dp[i][1]);
        dp[t][1] += dp[i][0];
    }
}
 
int main()
{
    //ifstream cin("in.txt");
    while(cin >> n && n )
    {
        memset(tree,0,sizeof(tree));
        memset(dp,0,sizeof(dp));
        map<string,int> cat;
        int ct = 1;
        cin >> str1;
        cat[str1] = ct ++;
        for(int i = 1;i < n;i ++)
        {
            cin >> str1 >> str2;
            if(cat[str1] == 0)  cat[str1] = ct ++;
            if(cat[str2] == 0)  cat[str2] = ct ++;
            tree[cat[str2]][cat[str1]] = 1;
        }
        fun(1);
        cout << max(dp[1][0],dp[1][1]) << ' ';
        int i,j;
        if(n == 2)  {cout << "No" << endl; continue;}
        for(i = 1;i <= n;i ++)
        if(dp[i][0] == dp[i][1])
        {
            for(j = 1;j <= n;j ++)
            {
                if(tree[i][j] && dp[j][0] == dp[j][1])
                break;
            }
            if(j <= n)  break;
        }
       if(i <= n) cout << "No" << endl;
       else cout << "Yes" << endl;
    }
    return 0;
}