hdu 4035 Maze 树形dp + 期望值

题目大意：

人需要逃离一棵树，每个节点有三种可能。

1.回到1节点，概率为k[i];

2.逃出树,概率为e[i];

3.逃到下一个临近的节点。(1-k[i]-e[i])

求逃离树的天数的期望值。

---

难点：

节点之间具有后续性。可以返回父亲节点。

---

题目分析：（节选自kungbin博客）

设 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则无解...

---

AC代码：

#include <bits/stdc++.h>
using namespace std;
#define N 10005
#define eps 1e-9
#define calcul(x) (1-k[x]-e[x])
double k[N],e[N];
double A[N],B[N],C[N];
vector<int>vec[N];
bool dfs(int t,int pre){
    int m = vec[t].size();
    A[t]=k[t];
    B[t]=(calcul(t))/m;
    C[t]=(calcul(t));
    double res = 0;
    for(int i=0;i<m;i++){
        int v=vec[t][i];
        if(v == pre)continue;
        if(!dfs(v,t))return false;
        A[t]+=calcul(t)/m*A[v];
        C[t]+=calcul(t)/m*C[v];
        res+=calcul(t)/m*B[v];
    }
    if(fabs(res-1)<eps)return false;
    A[t]/=(1-res);
    B[t]/=(1-res);
    C[t]/=(1-res);
    return true;
}
int main(){
    int T;
    scanf("%d",&T);
    int iCase = 0,n;
    while(T--){
        iCase++;
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            vec[i].clear();
        }
        for(int i=1;i<n;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        for(int i=1;i<=n;i++){
            scanf("%lf%lf",&k[i],&e[i]);
            k[i]/=100,e[i]/=100;
        }
        printf("Case %d: ",iCase);
        if(dfs(1,-1)&&fabs(1-A[1])>eps){
            printf("%.6f\n",C[1]/(1-A[1]));
        }else printf("impossible\n");
    }
}