2019南京网络赛 B robots

题意

给一个 $DAG$，一个机器人从 $1$ 走到 $n$ ，每一步会等概率走向某一个后继或不动，每一步（包括不动）的花费是当前的步数。求从起点走到终点的期望花费期望。
期望 DP。

分析

$f[u]$ 表示从 $u$ 出发的期望步数，$g[u]$ 表示从 $u$ 出发的期望花费。
即可列出方程：$$\{\begin{array}{l}f[u]=\frac{\sum _{(u,v)\in \text{DAG}}f[v]+f[u]}{d[u]+1}+1\\ g[u]=\frac{\sum _{(u,v)\in \text{DAG}}g[v]+g[u]+\sum _{(u,v)\in \text{DAG}}f[v]+f[u]}{d[u]+1}+1\end{array}$$
记忆化搜索分别解出所有的$f$和$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]);

    }
}