hdu2412:Party at Hali-Bula map应用&&树形DP

题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=2412

要注意的是:引入一个unique[n][2],其中unique[i][0]表示不选点i时选最多点情况是否唯一，unique[i][1]表示选点i时选最多情况是否唯一。

再就是处理好输入部分，就OK了。因为这我不知WA了多少次!!

废话不多说，给出我的 AC代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<map>
using namespace std;

const int maxn = 201;
int father[maxn];
int dp[maxn][2];
int uque[maxn][2];
bool visited[maxn];
bool flag;
int n;
int max(int a, int b){
    return a>b ? a : b;
}
void dfs(int x){
    visited[x] = true;
    int i;
	dp[x][0]=0;

    dp[x][1]=1;

    uque[x][0]=1;

    uque[x][1]=1;
    for (i = 0; i < n; i++){
        if (!visited[i] && father[i] == x)
        {
            visited[i] = true;
            dfs(i);
            dp[x][0] += max(dp[i][1], dp[i][0]);
            dp[x][1] += dp[i][0];
                if (dp[i][0]>dp[i][1]&&uque[i][0]==0){
                    uque[x][0] = 0;
                }
                else if (dp[i][0] < dp[i][1] && uque[i][1] == 0){
                    uque[x][0] = 0;
                }
                else if (dp[i][0] == dp[i][1]){
                    uque[x][0] = 0;
                }
                if (uque[i][0] == 0){
                    uque[x][1] = 0;
                }
        }
    }
}
void input(){
    map<string, int> mp;
    map<string, int>::iterator it,it2;
    string a, b;
    memset(father, -1, sizeof(father));
    cin >> a;
	int ans1,ans2;
    father[0] = 0;
    mp.insert(pair<string, int>(a, 0));
    int i;
	int t=1;
    for (i = 1; i < n; i++){
        cin >> a >> b;
		it=mp.find(a);
        it2 = mp.find(b);
		ans1=ans2=-1;
		if(it!=mp.end()){
			ans1=it->second;
		}
		if(it2!=mp.end()){
			ans2=it2->second;
		}
		if(ans1==-1){
			ans1=t;
			mp.insert(pair<string, int>(a, t));
			t++;
		}
		if(ans2==-1){
			ans2=t;
			mp.insert(pair<string, int>(b, t));
			t++;
		}
        father[ans1] = ans2;
        
    }
    memset(dp, 0, sizeof(dp));
    memset(uque, 0, sizeof(uque));
    memset(visited, false, sizeof(visited));
    dfs(0);
}
int main()
{
    while (cin >> n, n){
        flag = false;
        input();
        int ans = max(dp[0][0], dp[0][1]);
        cout << ans ;
        if (dp[0][0]>dp[0][1]&&uque[0][0]){
            cout << " Yes" << endl;
        }else if(dp[0][1]>dp[0][0]&&uque[0][1]){
			cout<<" Yes"<<endl;
		}
        else{
            cout << " No" << endl;
        }
    }
    return 0;
}