码迷,mamicode.com
首页 > 其他好文 > 详细

HDU-4035 Maze (概率DP求期望)

时间:2016-03-27 12:41:00      阅读:156      评论:0      收藏:0      [点我收藏+]

标签:

题目大意:在一个树形迷宫中,以房间为节点。有n间房间,每间房间存在陷阱的概率为ki,存在出口的概率为ei,如果这两种情况都不存在(概率为pi),那么只能做出选择走向下一个房间(包括可能会走向上一个房间)。根节点为1,当遇到陷阱时必须返回到根节点1处重新开始,当遇到出口时,走出迷宫。问从开始到走出迷宫所做出选择次数的期望值。

题目分析:定义状态dp(i)表示在节点 i 处直到走出迷宫的选择次数期望值。则状态转移方程为:

dp(i)=ki*dp(1)+(1/m)*pi*∑(dp(son)+1)  (i为叶子节点)  <1>

dp(i)=ki*dp(1)+(1/m)*pi*(dp(father)+1)+(1/m)*pi*∑(dp(son)+1)  (i为非叶子节点)  <2>

将<2>整理一下,得到:

dp(i)=ki*dp(1)+(1/m)*pi*dp(father)+(1/m)*pi*∑dp(son)+pi

显然,dp(i)都与dp(1)有关,另dp(i)=A(i)*dp(1)+B(i)*dp(father)+C(i)  <3>

将<3>带入∑dp(son),得到:

(1-(pi/m)*∑A(son))dp(i)=(ki+(pi/m)∑A(son))*dp(1)+(pi/m)*dp(father)+(pi/m)*∑C(son)+pi

显然,A(i),B(i),C(i)可得。

dp(1)=C(1)/(1-A(1)) 即为答案。

 

代码如下:

# include<iostream>
# include<cstdio>
# include<cmath>
# include<vector>
# include<cstring>
# include<algorithm>
using namespace std;

const int N=10005;
const int INF=100000;
const double eps=1e-9;

int n;
double k[N];
double e[N];
double A[N],B[N],C[N];
vector<int>G[N];

bool dfs(int u,int fa)
{
    int m=G[u].size();
    double temp=0;
    A[u]=k[u];
    B[u]=(1-k[u]-e[u])/m;
    C[u]=1-k[u]-e[u];
    for(int i=0;i<m;++i){
        int v=G[u][i];
        if(v==fa) continue;
        if(!dfs(v,u)) return false;
        A[u]+=(1-k[u]-e[u])*A[v]/m;
        C[u]+=(1-k[u]-e[u])*C[v]/m;
        temp+=(1-k[u]-e[u])*B[v]/m;
    }
    if(fabs(1-temp)<eps) return false;
    A[u]/=(1-temp);
    B[u]/=(1-temp);
    C[u]/=(1-temp);
    return true;
}

int main()
{
    int T,x,y,cas=0;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;++i)
            G[i].clear();
        for(int i=1;i<n;++i){
            scanf("%d%d",&x,&y);
            G[x].push_back(y);
            G[y].push_back(x);
        }
        for(int i=1;i<=n;++i){
            scanf("%lf%lf",k+i,e+i);
            k[i]/=100;
            e[i]/=100;
        }
        printf("Case %d: ",++cas);
        if(dfs(1,-1)&&fabs(1-A[1])>eps)
            printf("%.6lf\n",C[1]/(1-A[1]));
        else printf("impossible\n");
    }
    return 0;
}

  

HDU-4035 Maze (概率DP求期望)

标签:

原文地址:http://www.cnblogs.com/20143605--pcx/p/5325258.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!