洛谷 P6835 [Cnoi2020] 线形生物 题解

Announcement

• Programmed on 2024/2/25
• Written on 2024/2/25

题目来源

• 洛谷 P6835

Description

给定 $n+1$ 个节点，其中对于每个节点 $i\le n$，都有一条边 $i\to i+1$，同时给定 $m$ 条返祖边 $u\to v$（其中 $u\ge v$）。

求从 $1$ 号节点出发，每次等概率地沿着当前所在节点的出边去往下一个节点，最终到达 $n+1$ 号节点的期望步数。

• $1\le n,m\le 10^6$。

Solution

对于在图上随机游走的问题，有一个基于期望的线性性质的通用的技巧。

对于本题，若设 $E_{x\to y}$ 表示从 $x$ 到 $y$ 的期望步数，则 $E_{x\to y}=\sum _{i=x}^{y-1}{E}_{x\to x+1}$。

那么，只需求出所有 $i\in [1,n]\cap \mathbb{Z}$ 的 $E_{i\to i+1}$，答案即为 $\sum _{i=1}^n{E}_{i\to i+1}$。

为何是从 $1$ 开始求和呢？因为可能存在自环，$E_{1\to 1}$ 并非恒等于 $0$。

考虑如何计算 $E_{x\to x+1}$。

若设节点 $x$ 有 $de{g}_x$ 个返祖出边，$E$ 为返祖边的集合，$f_x=E_{x\to x+1}$，$su{m}_x=\sum _{i=1}^x{f}_x$。则根据期望的计算公式有：
$$\begin{array}{rl}{f}_x& =\frac{1}{de{g}_x+1}\cdot 1+\frac{1}{de{g}_x+1}\cdot \sum _{(x,y)\in E}(E_{y\to x+1}+1)\\ & =\frac{1}{de{g}_x+1}+\frac{1}{de{g}_x+1}\sum _{(x,y)\in E}(su{m}_x-su{m}_{y-1})+\frac{de{g}_x}{de{g}_x+1}\\ & =1+\frac{de{g}_x}{de{g}_x+1}{f}_x+\frac{1}{de{g}_x+1}\sum _{(x,y)\in E}(su{m}_{x-1}-su{m}_{y-1})\end{array}$$
将 $f_x$ 合并至等式左侧有：
$$\begin{array}{rl}\frac{1}{de{g}_x+1}{f}_x& =1+\frac{1}{de{g}_x+1}\sum _{(x,y)\in E}(su{m}_{x-1}-su{m}_{y-1})\\ f_x& =de{g}_x+1+\sum _{(x,y)\in E}(su{m}_{x-1}-su{m}_{y-1})\end{array}$$
从小到大依次计算 $f_x$ 即可，复杂度 $O(n+m)$。

Conclusion

• 对于在图上随机游走的问题，可将期望拆分成每一步的期望之和。

Code

#include <bits/stdc++.h>
using namespace std;
const int p=998244353;
int id,n,m,head[1000005],ver[1000005],nxt[1000005],tot,deg[1000005],f[1000005],sum[1000005]; // f[i] i->i+1的期望步数
void add(int u,int v){
	ver[++tot]=v,deg[u]++;
	nxt[tot]=head[u],head[u]=tot;
}
int main(){
	scanf("%d%d%d",&id,&n,&m);
	for (int i=1,u,v;i<=m;i++) scanf("%d%d",&u,&v),add(u,v);
	for (int i=1;i<=n;i++){
		f[i]=deg[i]+1;
		for (int j=head[i];j;j=nxt[j]){
			int u=ver[j];
			f[i]=((f[i]+sum[i-1]-sum[u-1])%p+p)%p;
		}
		sum[i]=(sum[i-1]+f[i])%p;
	}
	printf("%d\n",sum[n]);
	return 0;
}