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

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

要注意的是:引入一个unique
[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;
}