hdu 4035 Maze （树形概率dp）

题意：

给出一棵树，每个点都有两个概率，一个是回到原点的概率，一个是走出去的概率，然后人从原点（1）开始走。求：走出去之前走的步数的期望。

题解：

觉得这题很难，看了大牛的推导才做出来。

推导如下：转子kuangbin

设 E[i]表示在结点i处，要走出迷宫所要走的边数的期望。E[1]即为所求。

    叶子结点：
    E[i] = ki*E[1] + ei*0 + (1-ki-ei)*(E[father[i]] + 1);
         = ki*E[1] + (1-ki-ei)*E[father[i]] + (1-ki-ei);

    非叶子结点：（m为与结点相连的边数）
    E[i] = ki*E[1] + ei*0 + (1-ki-ei)/m*( E[father[i]]+1 + ∑( E[child[i]]+1 ) );
         = ki*E[1] + (1-ki-ei)/m*E[father[i]] + (1-ki-ei)/m*∑(E[child[i]]) + (1-ki-ei);

    设对每个结点：E[i] = Ai*E[1] + Bi*E[father[i]] + Ci;

    对于非叶子结点i，设j为i的孩子结点，则
    ∑(E[child[i]]) = ∑E[j]
                   = ∑(Aj*E[1] + Bj*E[father[j]] + Cj)
                   = ∑(Aj*E[1] + Bj*E[i] + Cj)
    带入上面的式子得
    (1 - (1-ki-ei)/m*∑Bj)*E[i] = (ki+(1-ki-ei)/m*∑Aj)*E[1] + (1-ki-ei)/m*E[father[i]] + (1-ki-ei) + (1-ki-ei)/m*∑Cj;
    由此可得
    Ai =        (ki+(1-ki-ei)/m*∑Aj)   / (1 - (1-ki-ei)/m*∑Bj);
    Bi =        (1-ki-ei)/m            / (1 - (1-ki-ei)/m*∑Bj);
    Ci = ( (1-ki-ei)+(1-ki-ei)/m*∑Cj ) / (1 - (1-ki-ei)/m*∑Bj);

    对于叶子结点
    Ai = ki;
    Bi = 1 - ki - ei;
    Ci = 1 - ki - ei;

    从叶子结点开始，直到算出 A1,B1,C1;

    E[1] = A1*E[1] + B1*0 + C1;
    所以
    E[1] = C1 / (1 - A1);
    若 A1趋近于1则无解...

题解如下：

技巧性比较大

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
//typedef __int64 lld;
#define oo 0x3f3f3f3f
#define Mod 1000000007
#define maxn 10005
#define Exp 1e-9
vector<int>G[maxn];
double A[maxn],B[maxn],C[maxn];
double p[maxn][2];

bool Ptree_dp(int u,int pre)
{
    double temp=0.0,m=G[u].size()*1.0,f=0;
    A[u]=p[u][0];
    B[u]=(1.0-p[u][0]-p[u][1])/m;//如果是叶子节点m=1所以就统一了代码
    C[u]=1.0-p[u][0]-p[u][1];
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(pre==v)continue;
        if(!Ptree_dp(v,u))return false;
        A[u]+=(1.0-p[u][0]-p[u][1])/m*A[v];
        C[u]+=(1.0-p[u][0]-p[u][1])/m*C[v];
        temp+=(1.0-p[u][0]-p[u][1])/m*B[v];
    }
    if(fabs(temp-1.0)<Exp)return false;
    A[u]/=(1.0-temp);
    B[u]/=(1.0-temp);
    C[u]/=(1.0-temp);
    return true;
}

int main()
{
    int n,m,T,u,v;
    scanf("%d",&T);
    for(int cas=1;cas<=T;cas++)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            G[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d %d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%lf %lf",&p[i][0],&p[i][1]);
            p[i][0]/=100.0;
            p[i][1]/=100.0;
        }
        printf("Case %d: ",cas);
        if(!Ptree_dp(1,-1)||fabs(1.0-A[1])<Exp)
            printf("impossible\n");
        else
            printf("%.6lf\n",C[1]/(1.0-A[1]));
    }
    return 0;
}