树形dp-poj3342 - 树的最大独立集 及其 是否唯一

题目链接：http://poj.org/problem?id=3342
题目大意：
思路：

字符串用map编个号。然后存树，可以无向图，也可以有向图，因为这里输入a b，b是a的老板。那么就是b->a。

此处用d[u][0]表示不选择节点u时，以u为根的子树的最大可选人数，那么d[u][1]则表示选择节点u的时候，以u为根的子树的最大可选人数。

同样的f[u][0]表示不选择节点u时，以u为根的子树的最大可选人数是否唯一，f[u][1]表示选择时的情况。

此时考虑状态转移方程：

1.选择节点u的时候，所有子节点必须不选d[u][1]=sum(d[v][0])+1 考虑独立性：只有在任意一个子节点f[v][0]不唯一的时候，f[u][1]就不唯一。

2.不选择节点u的时候，子节点可选可不选 d[u][0]=sum(max(d[v][0],d[v][1])) 考虑独立性：当d[v][0]和d[v][1]值相同的时候，选择哪一个都行，那么就不唯一了。若是选择的其中大的那个不唯一，那么f[u][0]也不唯一。

因为写错一个东西，从下午5点wa到晚上10点。

#define LL long long
using namespace std;

int dp[505][2];
int f[505][2];
vector<int> v[505];
map<string , int> mp;
int cut=0;
char a[305], b[305];
bool vis[305];

int ID(string n)
{
    if(mp.find(n)!=mp.end())
    {
        return mp[n];
    }
    else
    {
        mp[n]=++cut;
        return cut;
    }
}

int dfs(int u)
{
    vis[u]=1;
    if(dp[u][0]!=-1)
    {
        return 0;
    }
    dp[u][0]=dp[u][1]=0;
    for(int i=0;i<v[u].size();i++)
    {
        int to=v[u][i];
        if(vis[to])
        {
            continue;
        }
        dfs(to);//必须要求子节点已经求出

        dp[u][1]+=dp[to][0];
        f[u][1]=max(f[to][0], f[u][1]);

        //写成f[u][0]=(dp[to][0]==dp[to][1]?1:0),WA了一下午。
        f[u][0]=max(f[u][0], (dp[to][0]==dp[to][1]?1:0));
        if(dp[to][0]>dp[to][1])
        {
            dp[u][0]+=dp[to][0];
            f[u][0]=max(f[u][0], f[to][0]);
        }
        else
        {
            dp[u][0]+=dp[to][1];
            f[u][0]=max(f[u][0], f[to][1]);
        }

    }
    dp[u][1]++;
}

int main()
{
    int n;
    while(~scanf("%d",&n),n)
    {
        cut=0;
        mp.clear();
        memset(v, 0, sizeof(v));
        memset(dp, -1, sizeof(dp));
        memset(f, 0, sizeof(f));
        memset(vis, 0, sizeof(vis));

        scanf("%s",&a);
        mp[a]=++cut;
        for(int i=1;i<n;i++)
        {
            scanf("%s%s",a,b);
            v[ID(b)].push_back(ID(a));
            v[ID(a)].push_back(ID(b));
        }

        dfs(1);
        int flag=max(dp[1][0]==dp[1][1]?1:0, ((dp[1][0]>dp[1][1])?f[1][0]:f[1][1]));//判断根节点
        cout<<max(dp[1][0], dp[1][1])<<" "<<(flag?"No":"Yes")<<endl;
    }

    return 0;
}