UVA 1220 树形dp 最大独立集+状态标记

题意：公司里n个人形成一个树形结构，即除了boss之外每个员工都有唯一的直属上司。要求选尽量多的人，但不能同时选择一个人和他的直属上司。问最多能选多少人，以及判断人数最多的前提下得方案是否是唯一结果。

思路：树形dp，就是求在树上的最大独立集合，加唯一性特判.

首先需要明确的一点是： 这是一个树形结构。

A：然后就是要明确 我们要存入两种结果值，一个是最大的独立集合的值， 另一个是判断该结果是否是唯一的。

所以 最好要用两个数组进行存储。   dp      与      f

B : 然后再明确数组的下标应该为啥 ： 

     a：对于最大独立集合的值来说，相对于一个点  存在 dp(u) =max(1+sum（dp[j] ，j属于 u的孙子们)，sum（dp[ j ], j属于u的儿子们））；

     b：对于一个结果是否唯一性；

    如何将这两种结果结合起来呢！

      dp[ u, 0 ] ,f[ u, 0 ]表示以u为根的子树中，不选u能得到的最大人数以及方案唯一性（f[ u, 0 ]=1表示唯一，=0 表示不唯一）；

       dp[ u, 1], f[ u,1 ]表示以u为根的子树中，选u能得到的最大人数以及方案唯一性

C：对于dp（u，1）的计算因为选了u，所以u

       dp[u][1]+=dp[v][0];   并且对于存在f[v][0]==0 就说明f[u][1]=0;

      对于 对于dp（u，0）的计算因为不选u，

       dp[u][0]+=max(dp[v][1],dp[v][0]);
        if(dp[v][0]>dp[v][1]&&f[v][0]==0)  f[u][0]=0;
        else if(dp[v][1]>dp[v][0]&&f[v][1]==0) f[u][0]=0;
        else if (dp[v][0]==dp[v][1]) f[u][0]=0;

详解见紫书 283页；

 

#include<bits/stdc++.h>
using namespace std;
const int N= 250;
int dp[N][2];
int f[N][2];
map<string,int>mp;
vector<int>vec[N];
char s1[N*100],s2[N*100];
void dfs(int u){
    dp[u][0]=0; dp[u][1]=1;
    f[u][0]=1; f[u][1]=1;
    int len=vec[u].size();
    for(int i=0;i<len;i++){
        int v=vec[u][i];
        dfs(v);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][1],dp[v][0]);
        if(dp[v][0]>dp[v][1]&&f[v][0]==0)  f[u][0]=0;
        else if(dp[v][1]>dp[v][0]&&f[v][1]==0) f[u][0]=0;
        else if (dp[v][0]==dp[v][1]) f[u][0]=0;
        if(f[v][0]==0) f[u][1]=0;
    }
}
int main(){
    int n;
    while(~scanf("%d",&n),n){
        memset(f,0,sizeof(f));
        memset(dp,0,sizeof(dp));
        memset(vec,0,sizeof(vec));
        mp.clear();
        int top=0;
        for(int i=0;i<n;i++){
            scanf("%s",s1);
            if(!mp.count(s1))
                mp[s1]=++top;
            if(i==0)continue;
            scanf("%s",s2);
            if(mp.count(s2)==0)
                mp[s2]=++top;
            vec[mp[s2]].push_back(mp[s1]);
        }
        dfs(1);
        if(dp[1][1]>dp[1][0]&&f[1][1]) printf("%d Yes\n",dp[1][1]);
        else if(dp[1][0]>dp[1][1]&&f[1][0]) printf("%d Yes\n",dp[1][0]);
        else printf("%d No\n",max(dp[1][1],dp[1][0]));
    }
    return 0;
}