The Lost House

POJ

题意:(n(n<=1000))个节点的树,蜗牛把壳落在了一个树叶上,壳在每一个树叶上的概率是相同的.现在蜗牛从树根开始爬,在树杈中可能会有毛毛虫,毛毛虫会告诉蜗牛壳是否在这个点的子树中.每个树枝的长度为1,求蜗牛最终能找到壳需要爬行的距离的期望值最小是多少.

分析:本人题解结合了博客一的讲解和博客二的实现.

期望值=(sum)蜗牛到第i个叶子的距离(*)壳在第i个叶子上的概率= 蜗牛到所有叶子节点的距离的和/叶子节点数.于是题目转化为了,我们要求一个序列,按这个序列到达每一个叶子节点的距离的和是最小的.

设节点x,y都是节点u的子节点,假设壳在u的子树中(即壳在子树x或者子树y中).那么我们有两种选择:

选择一:先走x,再走y.则期望值=在x上找到壳的期望步数壳在x上的概率+(在x上寻找失败后由x返回u的步数+在y上找到壳的期望步数)壳在y上的概率

选择二:先走y,再走x.则期望值=在y上找到壳的期望步数壳在y上的概率+(在y上寻找失败后由y返回u的步数+在x上找到壳的期望步数)壳在x上的概率

拆开括号,可以发现两个选择,即两个式子只有一个地方不同,选择一是 在x上寻找失败后由x返回u的步数壳在y的概率;选择二是 在y上寻找失败后由y返回u的步数壳在x上的概率

所以我们只要依据这个从小到大排序,就能得到我们是先走x,还是先走y了.

以上是关于如何走的讨论,可以说是本题的核心部分.下面讲如何实现.

(size[u])表示以u为根的子树中叶子节点的个数,(su[u])表示在以u为根的子树中找到壳期望走的距离(1为根节点,(su[1]/size[1])即为最终答案),设(fail[u])表示在以u为根的子树中,遍历整个子树都没有找到壳又返回x节点期望走的距离.

设v是u的子节点,dfs回溯的时候(size[u]+=size[v])就可以处理出size数组了.(su[u]+=(fail[u]+1)*size[v]+su[v])表示当前考虑u的子树,之前遍历的子树一直没找到壳,现在走进了子树v找到了壳;(fail[u]+= fail[v]+2)表示遍历了子树v没有找到壳,所以返回节点u.

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#define ll long long
using namespace std;
inline int read(){
    int x=0,o=1;char ch=getchar();
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')o=-1,ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*o;
}
const int N=1005;
int bj[N],size[N],su[N],fail[N];
vector<int>q[N];
inline bool cmp(int x,int y){
	return (fail[x]+2)*size[y]<(fail[y]+2)*size[x];
}
inline void dfs(int u){
	if(!q[u].size()){//如果是叶子节点
		size[u]=1;
		su[u]=0;fail[u]=0;
		return;
	}
	for(int i=0;i<q[u].size();++i){//遍历子节点
		int v=q[u][i];
		dfs(v);size[u]+=size[v];
	}
	sort(q[u].begin(),q[u].end(),cmp);//按照上面分析的那样排序
	for(int i=0;i<q[u].size();++i){
		int v=q[u][i];
		su[u]+=(fail[u]+1)*size[v]+su[v];
		fail[u]+=fail[v]+2;
	}
	if(bj[u])fail[u]=0;
//别忘了!!!如果有毛毛虫,就不会出现 走进去 又没找到壳 的情况
}
int main(){
    //freopen("D.in","r",stdin);
    //freopen("D.out","w",stdout);
	while(1){
		int n=read();if(!n)break;
		for(int i=1;i<=n;++i){
			bj[i]=0;size[i]=0;
			su[i]=0;fail[i]=0;
			q[i].clear();
		}//多组数据初始化
		int x=read();char ch;cin>>ch;//第一组没用
		for(int i=2;i<=n;++i){
			x=read();cin>>ch;
			if(ch=='Y')bj[i]=1;//如果有毛毛虫,打个标记
			q[x].push_back(i);//直接vector存图
		}
		dfs(1);printf("%.4f
",1.0*su[1]/size[1]);
	}
    return 0;
}

原文地址:https://www.cnblogs.com/PPXppx/p/11437289.html