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\n",1.0*su[1]/size[1]);
    }
    return 0;
}
上一篇:CSP2019: Lost and Found


下一篇:二分,倍增的一些思考(浅谈二分)(lost my music:可持久化栈)