poj 2057 树形dp

主要是在策略选择上：对于一个节点，它该先往哪个子节点走

比较它的两个子节点u和v：房子在u这边的概率为num[u]/ num[1] ，房子在v这边的概率为 num[v]/num[1]，分别简化为num[u] ,num[v]

先选u：num[u]*go[u] + num[v]*(fail[u]+2+go[v])

先选v：num[v]*go[v] + num[u]*(fail[v]+2+go[u])

化简可知 ，只要比较 num[v]*(fail[u]+2) ，num[u]*(fail[v]+2) 就好，选步数小的那种

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define N 1050
vector<int>vec[N];
int w[N],n,go[N],fail[N],num[N];

bool cmp(int u,int v)
{
	return (fail[u]+2)*num[v]<(fail[v]+2)*num[u];
}
void dfs(int u,int pre)
{
	int tt[30];
	int cnt=0;
	for(int i=0;i<vec[u].size();++i)
		if(vec[u][i]!=pre)
		{
			tt[cnt++]=vec[u][i];
			dfs(vec[u][i],u);
			fail[u]+=fail[vec[u][i]]+2;
			num[u]+=num[vec[u][i]];
		}
		// leaf node
	if(fail[u]==0 )
	{
		num[u]=1;
		return;
	}
	sort(tt,tt+cnt,cmp);
	fail[u]=0;
	for(int i=0;i<cnt;++i)
	{
		go[u]+=(fail[u]+1)*num[tt[i]]+go[tt[i]];
		fail[u]+=fail[tt[i]]+2;
	}
	if(w[u])
		fail[u]=0;
}
int main ()
{
	while(scanf("%d",&n)!=EOF)
	{
		if(n==0)break;
		memset(w,0,sizeof(w));
		memset(fail,0,sizeof(fail));
		memset(go,0,sizeof(go));
		memset(num,0,sizeof(num));
		for(int i=1;i<=n;++i)
			vec[i].clear();
		int u;char ch;
		for(int i=1;i<=n;++i)
		{
			scanf("%d",&u);
			scanf(" %c",&ch);
			if(u>0)
			{
				vec[i].push_back(u);
				vec[u].push_back(i);
			}
			if(ch=='Y')
				w[i]=1;
		}
		dfs(1,-1);
		double ans=1.0*go[1]/num[1];
		printf("%.4lf\n",ans);
	}
	return 0;
}