hdu2412 Party at HaliBula

http://acm.hdu.edu.cn/showproblem.php?pid=2412

树状DP，找出树中互不相邻的点，最多能有多少个，判断是否唯一比较麻烦

不涂即为白色，黑色表示选择该点

不唯一的涂色方法可以分为以下两种树：

1.(如果根节点为白色，最多的涂色方法)dp[i][0]  有多种的树

2.(如果根节点为黑色，最多的涂色方法)dp[i][1]  有多种的树

形成1类型树方法：

如果根节点为白色，它的子树中有任意一个满足下面条件：

dp[j][0] == dp[j][1] 

或 （dp[j][0] > dp[j][1] 且 该子树为1类型树）

或 （dp[j][1] > dp[j][0] 且 该子树为2类型树）

形成2类型树方法：

如果根节点为白色，它的子树中有任意一个是1类型树：

#include <stdio.h>
#include <iostream>
#include <string>
#include <vector>
#include <map>

using namespace std;

vector<int> p[210];
vector<int>::iterator it;
int n, dp[210][2], mark[210][2];

void dfs(int x)
{
    int i, j, sum1, sum2;
    if(p[x].empty())
    {
        dp[x][0] = 0;
        dp[x][1] = 1;
        mark[x][0] = mark[x][1] = 0;
        return;
    }
    sum1 = 0; //don't choose
    sum2 = 1; //choose
    for(i=0; i<p[x].size(); i++)
    {
        j = p[x][i];
        dfs(j);
        if(dp[j][1] > dp[j][0] && mark[j][1])
        {
            mark[x][0] = 1;
        }
        if(dp[j][0] > dp[j][1] && mark[j][0])
        {
            mark[x][0] = 1;
        }
        if(dp[j][0] == dp[j][1])
        {
            mark[x][0] = 1;
        }
        if(mark[j][0])
        {
            mark[x][1] = 1;
        }
        sum1 += max(dp[j][0], dp[j][1]);
        sum2 += dp[j][0];
    }
    dp[x][0] = sum1;
    dp[x][1] = sum2;
}

int main()
{
    string s, s1, s2;
    map<string, int> map1;
    int i, temp1, temp2;
    while(scanf("%d", &n), n)
    {
        for(i=1; i<=n; i++)
        {
            mark[i][0] = mark[i][1] = 0;
            p[i].clear();
        }
        map1.clear();
        cin >> s;
        map1.insert(make_pair(s, 1));
        i = 2;
        n = n - 1;
        while(n-- && cin >> s1 >> s2)
        {
            if(map1.find(s1) == map1.end())
            {
                map1.insert(make_pair(s1, i));
                i = i + 1;
            }
            if(map1.find(s2) == map1.end())
            {
                map1.insert(make_pair(s2, i));
                i = i + 1;
            }
            temp1 = map1[s1];
            temp2 = map1[s2];
            p[temp2].push_back(temp1);
        }
        dfs(1);
        if(dp[1][0] == dp[1][1])
        {
            printf("%d No\n", dp[1][0]);
            continue;
        }
        if(dp[1][0] > dp[1][1])
        {
            printf("%d ", dp[1][0]);
            printf(mark[1][0]? "No\n": "Yes\n");
            continue;
        }
        if(dp[1][0] < dp[1][1])
        {
            printf("%d ", dp[1][1]);
            printf(mark[1][1]? "No\n": "Yes\n");
        }
    }
    return 0;
}