hdu4035 Maze

链接

http://acm.hdu.edu.cn/showproblem.php?pid=4035

题解

就概率$dp$嘛
$f[i]$表示从$i$点出发走出迷宫的期望步数

$$f_i=k_if_1+\sum_{j}\frac{1-k_i-e_i}{d_i}f_j$$
其中$j$是枚举了和$i$相邻的点
那么显然我对于$n$个节点可以列出$n$个方程，这$n$个方程就能解出$f_1...f_n$，$f_1$就是答案
高斯消元的复杂度是$O(n^3)$，不能用
这个结构比较奇特，它是树形结构，之前我做过一道环状转移的题：
https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/81751128
环状直接设$f_1$为变量，去递推其它的系数就行，但是树形不行，因为你会发现$f_i$的式子里含有$f_j$，$f_j$的式子里又含有$f_i$，如果我想用$f_1$表示$f_i$，我得先计算$f_j$用$f_1$怎么表示，这样就陷入了无尽的迭代
那就这样，我用$f_j$和$f_1$表示$f_i$，那么就能推出$f_j$等于多少$f_1$了
那现在考虑下顺序问题，显然我要$dfs$遍历树，叶子节点和父亲组合成一组，我就能表示出父亲节点等于儿子的多少倍+父亲的多少倍+常数
这样一直向上回溯，相当于每次消掉当前点，最终得到只含有$f_1$的式子，从而解出答案

代码

//树形概率DP 
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cctype>
#define cl(x) memset(x,0,sizeof(x))
#define maxn 20100
#define eps 1e-8
#define inf (1e100)
using namespace std;
int N, tot, nex[maxn], head[maxn], to[maxn], d[maxn];
double A[maxn], B[maxn], C[maxn], k[maxn], e[maxn];
void adde(int a, int b){to[++tot]=b;nex[tot]=head[a];head[a]=tot;}
void init()
{
    cl(head), cl(to), cl(d), tot=0;
    int i, a, b;
    scanf("%d",&N);
    for(i=1;i<N;i++)scanf("%d%d",&a,&b), adde(a,b), adde(b,a), d[a]++, d[b]++;
    for(i=1;i<=N;i++)scanf("%lf%lf",k+i,e+i), k[i]/=100.0, e[i]/=100.0;
}
void dfs(int pos, int pre)
{
    int p;
    double sa=0, sb=0, sc=0, t;
    for(p=head[pos];p;p=nex[p])
        if(to[p]^pre)
        {
            dfs(to[p],pos);
            sa+=A[to[p]];
            sb+=B[to[p]];
            sc+=C[to[p]]+1;
        }
    if(pos!=1)
    {
        t=1-(1-k[pos]-e[pos])/d[pos]*sb;
        A[pos]=(k[pos]+(1-k[pos]-e[pos])/d[pos]*sa)/t;
        B[pos]=(1-e[pos]-k[pos])/d[pos]/t;
        C[pos]=((1-e[pos]-k[pos])/d[pos]*(sc+1))/t;
    }
    else
    {
        t=(1-(sa+sb)/d[1]);
        A[1]=0;
        C[1]=sc/d[1]/t;
    }
}
int main()
{
    int T, c;
    double ans;
    scanf("%d",&T);
    for(c=1;c<=T;c++)
    {
        init();
        dfs(1,0);
        printf("Case %d: ",c);
        ans=C[1];
        if(ans>inf)printf("impossible\n");
        else printf("%.10lf\n",ans);
    }
    return 0;
}