hdu4035 Maze 【期望dp + 数学】

时间:2022-11-04 09:59:04

题目链接

BZOJ4035

题解

神题啊。。。orz

不过网上题解好难看,数学推导不写\(Latex\)怎么看。。【Latex中毒晚期】

我们由题当然能很快写出\(dp\)方程

设\(f[i]\)表示从\(u\)出发逃离的期望步数,\(m\)为该点度数

\[\begin{aligned}
f[u] &= K_uf[1] + \frac{1 - K_u - E_u}{m}\sum\limits_{(u,v) \in edge} (f[v] + 1)\\
&= K_uf[1] + \frac{1 - K_u - E_u}{m}f[fa[u]] + \frac{1 - K_u - E_u}{m}\sum\limits_{(u,v) \in edge \& v \ne fa[u]} f[v] + (1 - K_u - E_u)\\
\end{aligned}
\]

然后就会发现这个方程似乎有后效性,立即想高斯消元

一看范围\(n \le 10^4\)什么鬼嘛QAQ。。。

题解是这么说的:

我们设

\[f[u] = A_uf[1] + B_uf[fa[u]] + C_u
\]

对于叶子节点,显然有

\[\begin{aligned}
A_u &= K_u \\
B_u &= 1 - K_u - E_u \\
C_u &= 1 - K_u - E_u \\
\end{aligned}
\]

对于非叶节点,我们展开\(f[v]\)

\[\begin{aligned}
f[u] &= K_uf[1] + \frac{1 - K_u - E_u}{m}f[fa[u]] + \frac{1 - K_u - E_u}{m}\sum\limits_{(u,v) \in edge \& v \ne fa[u]} f[v] + (1 - K_u - E_u)\\
&= K_uf[1] + \frac{1 - K_u - E_u}{m}f[fa[u]] + \frac{1 - K_u - E_u}{m}\sum\limits_{(u,v) \in edge \& v \ne fa[u]} (A_vf[1] + B_vf[u] + C_v) + (1 - K_u - E_u)\\
\end{aligned}
\]

我们整理一下:

\[f[u] = \frac{K_u + \frac{1 - K_u - E_u}{m}\sum A_v}{1 - \frac{1 - K_u - E_u}{m}\sum B_v}f[1] + \frac{\frac{1 - K_u - E_u}{m}}{1 - \frac{1 - K_u - E_u}{m}\sum B_v}f[fa[u]] + \frac{1 - K_u - E_u - \frac{1 - K_u - E_u}{m}\sum C_v}{1 - \frac{1 - K_u - E_u}{m}\sum B_v}
\]

\[\begin{aligned}
A_u &= \frac{K_u + \frac{1 - K_u - E_u}{m}\sum A_v}{1 - \frac{1 - K_u - E_u}{m}\sum B_v} \\
B_u &= \frac{\frac{1 - K_u - E_u}{m}}{1 - \frac{1 - K_u - E_u}{m}\sum B_v} \\
C_u &= \frac{1 - K_u - E_u - \frac{1 - K_u - E_u}{m}\sum C_v}{1 - \frac{1 - K_u - E_u}{m}\sum B_v} \\
\end{aligned}
\]

然后由于

\[\begin{aligned}
f[1] &= A_1f[1] + B_1 \times 0 + C_1 \\
f[1] &= \frac{C_1}{1 - A_1}
\end{aligned}
\]

当\(1 - A_1 = 0\)时无解

否则我们能直接计算出\(f[1]\),即为所求

是不是很神奇?

这个式子的推导主要是利用了式子中有\(f[fa[u]]\)这一项,从而可以从儿子中递推出父亲的信息

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define mp(a,b) make_pair<int,int>(a,b)
#define cls(s) memset(s,0,sizeof(s))
#define cp pair<int,int>
#define LL long long int
#define eps 1e-10
using namespace std;
const int maxn = 10005,maxm = 100005,INF = 1000000000;
inline int read(){
int out = 0,flag = 1; char c = getchar();
while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
int n,h[maxn],ne,de[maxn],fa[maxn];
struct EDGE{int to,nxt;}ed[maxn << 1];
void build(int u,int v){
ed[++ne] = (EDGE){v,h[u]}; h[u] = ne;
ed[++ne] = (EDGE){u,h[v]}; h[v] = ne;
de[u]++; de[v]++;
}
double A[maxn],B[maxn],C[maxn],K[maxn],E[maxn];
int dfs(int u){
if (de[u] == 1 && u != 1){
A[u] = K[u];
B[u] = C[u] = 1 - K[u] - E[u];
return true;
}
double m = de[u],tmp = 0;
A[u] = K[u];
B[u] = (1 - K[u] - E[u]) / m;
C[u] = 1 - K[u] - E[u];
Redge(u) if ((to = ed[k].to) != fa[u]){
fa[to] = u; if (!dfs(to)) return false;
A[u] += A[to] * B[u];
C[u] += C[to] * B[u];
tmp += B[to] * B[u];
}
if (fabs(1 - tmp) < eps) return false;
A[u] /= (1 - tmp); B[u] /= (1 - tmp); C[u] /= (1 - tmp);
return true;
}
int main(){
int T = read();
for (int t = 1; t <= T; t++){
n = read(); cls(de); cls(h); ne = 1;
for (int i = 1; i < n; i++) build(read(),read());
for (int i = 1; i <= n; i++) K[i] = read() / 100.0,E[i] = read() / 100.0;
printf("Case %d: ",t);
if (!dfs(1) || fabs(A[1] - 1) < eps) puts("impossible");
else printf("%.10lf\n",C[1] / (1 - A[1]));
}
return 0;
}