洛谷-游走-(期望dp+拓扑)

P6154

题意：
就是给你一个n点m边的有向图无环图，可能存在重边，然后小A会随机选择一条路径，都是等概率的，路径的起点和终点可以相同。问你小A走的路径长度的期望，同时对998244353取模。

思考：

1. 既然选择每条路径的概率都是相同的，那么我可以先把概率求出来，那么就是合法路径的总个数。这个可以跑一边拓扑序，把每个点当作起点的时候有多少条路径可以走。累加起来就是总路径个数。
2. 那么现在就是求路径的总长度了，对于这种求很多条路径的总长度，都是看每条边的贡献。所以这里就想到枚举所有边a->b，那么这条边的贡献就是：以a为终点的路径个数*以b为起点的路径个数。所以对图建一个返图，再跑一遍dp就可以了。
3. 当时做的时候，我以为是求到a点有多少点，a可以到多少点。按可达性统计那题做的。但是我发现边的贡献不仅仅是看点，而是看路径。对于可达性统计只能用bitset。如果求一个点可以走的路径那么就是拓扑dp，或者直接记忆化搜索也行。
4. 当然还有一种做法就是直接一遍拓扑dp，就是定义dp1为以i为终点的路径条数，dp2为以i为终点的走的路径条数。那么转移dp1[spot] = dp1[spot]+dp1[now]。dp2[spot] = dp2[spot]+dp2[now]+dp1[now]。这里为什么加dp1[now]是因为：到now点有这些路径，那么从now到spot，这条边肯定也走了那么多次。
5. 多多取模，不要忘记任何加的地方都取模。

代码：

两遍记忆化搜索
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 1e6+10,M = 2010;

int T,n,m,k;
PII va[N];
int in[N],out[N];
int vis[N];

vector<int > e1[N],e2[N];

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

int dfs1(int now)
{
	if(out[now]) return out[now];
	vis[now] = 1;
	out[now] = 1;
	for(auto spot:e1[now]) out[now] = (out[now]+dfs1(spot))%mod;
	return out[now];
}

int dfs2(int now)
{
	if(in[now]) return in[now];
	vis[now] = 1;
	in[now] = 1;
	for(auto spot:e2[now]) in[now] = (in[now]+dfs2(spot))%mod;
	return in[now];
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		va[i] = {a,b};
		e1[a].pb(b);
		e2[b].pb(a);
	}
	for(int i=1;i<=n;i++)
	{
		if(!vis[i]) dfs1(i);
	}
	for(int i=1;i<=n;i++) vis[i] = 0;
	for(int i=1;i<=n;i++)
	{
		if(!vis[i]) dfs2(i);
	}
	int ans = 0;
	for(int i=1;i<=m;i++)
	{
		ans = (ans+in[va[i].fi]*out[va[i].se]%mod)%mod;
	}
	int sum = 0;
	for(int i=1;i<=n;i++)
	{
		sum = (sum+out[i])%mod;
	}
	ans = ans*ksm(sum,mod-2)%mod;
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}
两遍拓扑
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 1e6+10,M = 2010;

int T,n,m,k;
PII va[N];
int dp1[N],dp2[N];
int in1[N],in2[N];

vector<int > e1[N],e2[N];

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

void topsort1()
{
	queue<int > q;
	for(int i=1;i<=n;i++)
	{
		dp1[i] = 1;
		if(!in1[i]) q.push(i);
	}
	while(q.size())
	{
		auto now = q.front();
		q.pop();
		for(auto spot:e1[now])
		{
			dp1[spot] = (dp1[spot]+dp1[now])%mod;
			if(--in1[spot]==0) q.push(spot);
		}
	}
}

void topsort2()
{
	queue<int > q;
	for(int i=1;i<=n;i++)
	{
		dp2[i] = 1;
		if(!in2[i]) q.push(i);
	}
	while(q.size())
	{
		auto now = q.front();
		q.pop();
		for(auto spot:e2[now])
		{
			dp2[spot] = (dp2[spot]+dp2[now])%mod;
			if(--in2[spot]==0) q.push(spot);
		}
	}
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		va[i] = {a,b};
		e1[a].pb(b);
		e2[b].pb(a);
		in1[b]++;in2[a]++;
	}
	topsort1();
	topsort2();
	int ans = 0,sum = 0;
	for(int i=1;i<=m;i++) ans = (ans+dp1[va[i].fi]*dp2[va[i].se]%mod)%mod;
	for(int i=1;i<=n;i++) sum = (sum+dp2[i])%mod;
	ans = ans*ksm(sum,mod-2)%mod;
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}
一遍拓扑
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 1e6+10,M = 2010;

int T,n,m,k;
PII va[N];
int dp1[N],dp2[N];
int in[N];

vector<int > e[N];

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

void topsort()
{
	queue<int > q;
	for(int i=1;i<=n;i++)
	{
		dp1[i] = 1;
		if(!in[i]) q.push(i);
	}
	while(q.size())
	{
		auto now = q.front();
		q.pop();
		for(auto spot:e[now])
		{
			dp2[spot] = (dp2[spot]+dp2[now]+dp1[now])%mod;
			dp1[spot] = (dp1[spot]+dp1[now])%mod;
			if(--in[spot]==0) q.push(spot);
		}
	}
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		in[b]++;
	}
	topsort();
	int ans = 0,sum = 0;
	for(int i=1;i<=n;i++)
	{
		ans = (ans+dp2[i])%mod;
		sum = (sum+dp1[i])%mod;
	}
	ans = ans*ksm(sum,mod-2)%mod;
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}

总结：
注意细节。