题意
给一个
D
A
G
DAG
DAG,一个机器人从
1
1
1 走到
n
n
n ,每一步会等概率走向某一个后继或不动,每一步(包括不动)的花费是当前的步数。求从起点走到终点的期望花费期望。
期望 DP。
分析
f
[
u
]
f[u]
f[u] 表示从
u
u
u 出发的期望步数,
g
[
u
]
g[u]
g[u] 表示从
u
u
u 出发的期望花费。
即可列出方程:
{
f
[
u
]
=
∑
(
u
,
v
)
∈
DAG
f
[
v
]
+
f
[
u
]
d
[
u
]
+
1
+
1
g
[
u
]
=
∑
(
u
,
v
)
∈
DAG
g
[
v
]
+
g
[
u
]
+
∑
(
u
,
v
)
∈
DAG
f
[
v
]
+
f
[
u
]
d
[
u
]
+
1
+
1
\begin{cases} f[u]=\frac{ \sum \limits_{(u,v)\in \text{DAG}}f[v] + f[u] } {d[u] + 1} + 1 \\ g[u] = \frac{\sum\limits_{(u,v)\in \text{DAG}}g[v] + g[u]+\sum\limits_{(u,v) \in \text{DAG}}f[v] + f[u]}{d[u] + 1} + 1\end{cases}
⎩⎪⎨⎪⎧f[u]=d[u]+1(u,v)∈DAG∑f[v]+f[u]+1g[u]=d[u]+1(u,v)∈DAG∑g[v]+g[u]+(u,v)∈DAG∑f[v]+f[u]+1
记忆化搜索分别解出所有的
f
f
f和
g
g
g即可。
代码
#include<bits/stdc++.h>
using namespace std;
typedef pair<double, double> P;
const int maxn = 4e5 + 10;
int out[maxn], n, m;
double f[maxn], g[maxn];
vector<int> G[maxn];
void add_edge(int u, int v)
{
G[u].push_back(v);
}
void dfs(int u)
{
if (u == n) return;
double sumf = 0, sumg = 0;
for (int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
if (f[v] == 0) dfs(v);
sumf += f[v];
sumg += g[v];
}
f[u] = (out[u] + 1 + sumf) / out[u];
g[u] = (out[u] + 1 + sumg + sumf + f[u]) / out[u];
}
void init()
{
memset(f, 0, sizeof(f));
memset(g, 0, sizeof(g));
memset(out, 0, sizeof(out));
for (int i = 0; i < maxn; i++)
G[i].clear();
}
int main()
{
int t;
scanf("%d", &t);
while (t--) {
init();
scanf("%d%d", &n, &m);
for (int i = 0; i < m; i++) {
int u, v;
scanf("%d%d", &u, &v);
add_edge(u, v);
out[u]++;
}
dfs(1);
printf("%.2f\n", g[1]);
}
}