poj 3342 Party at Hali-Bula 树形dp

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#include <cstring>
using namespace std;
const int maxn= 310;
struct node{
int v, next;
}edge[maxn];
int dp[maxn][2], vis[maxn], pre[maxn], value, n, flag;
map<string, int> mp;
void add( int aa, int bb){
edge[value].v= aa;
edge[value].next= pre[bb];
pre[bb]= value++;
}
void init(){
int i, j, k, cnt= 1, aa, bb;
char tmp1[110], tmp2[110];
mp.clear();                  // STL 全局变量 记着clear......................
memset( vis, 0, sizeof( vis));
memset( dp, 0, sizeof( dp));
memset( pre, -1, sizeof( pre));
value= 1;
scanf("%s", tmp1);
mp[tmp1]= cnt++;
for( i= 1; i<n; i++){
scanf("%s%s", tmp1, tmp2);
if( !mp[tmp1]) mp[tmp1]= cnt++;
if( !mp[tmp2]) mp[tmp2]= cnt++;
aa= mp[tmp1];
bb= mp[tmp2];
add(aa, bb);
}
}
int MAX(int x, int y){
return x>y?x:y;
}
void dfs(int u){
int i, j, k, p;
if( vis[u]) return;
vis[u]= 1;
for( j= pre[u]; j!= -1; j= edge[j].next){
p= edge[j].v;
dfs(p);
dp[u][1]+= dp[p][0];
dp[u][0]+= MAX( dp[p][0], dp[p][1]);
}
dp[u][1]++;
}
int check(){
int i, j, k;
for( i=1; i<=n; i++){
//注意1节点，为根，没有父节点
//所以要考虑其dp[1][1]== dp[1][0]情况时，需特殊处理
//在main函数中先判断了dp[1][1]== dp[1][0]，才觉得要不要进入check函数
if( dp[i][0] >= dp[i][1]){
//对于某个节点，当满足此种情况时，应选择dp[i][0],
//而dp[u][0]为的累加和 MAX( dp[p][0], dp[p][1])
//若dp[p][0]==dp[p][1]时，即子节点选与不选结果相同，即不唯一
for( j= pre[i]; j!= -1; j= edge[j].next){
if( dp[edge[j].v][0] == dp[edge[j].v][1])
return 0;
}
}
}
return 1;
}
int main(){
freopen("1.txt", "r", stdin);
int ans, i;
while( scanf("%d", &n) && n){
init();
dfs(1);
flag = 1;
if( dp[1][0]== dp[1][1] )
flag= 0;
if( flag) flag= check();
printf("%d ", MAX(dp[1][0], dp[1][1]));
if( flag )printf("Yes\n");
else printf("No\n");
}
return 0;
}