牛客多校-Two Frogs-(期望dp+枚举方式优化dp)

B

题意：
就是现在有n个荷叶，你在第i个荷叶上可以跳到[i+1,i+va[i]]这段区间的荷叶上。当你跳到最后一个荷叶上就结束了。现在小A随机的跳，小B也随机的跳，问你他俩跳到n号荷叶上用的相同步数的期望。

思考：

1. 首先既然小A和小B都是随机的，那么他俩的概率和期望步数都一样。刚开始我以为是求步数的期望，那么E[a] = p[a]×[a]×a。那么可以先求概率，或者直接求出来一个人的E1[a] = p[a]×a。那么两个人的E[a] = E1[a]×E1[a]/a就可以了。但是求出来发现不对，其实是是否相同的期望，也就是如果相同就是1，不相同就是0。那么就是求走到n点i步的概率平方之和。
2. 然后我就直接dp[i][j]为走到i点走j步的概率，但是这样是三重循环，第一重走到哪个荷叶，第二重为步数，第三重是这个i点可以从前面的哪些点转移。这里就卡了会，我总感觉这种做法太经典了，能优化，就像之前做的这个：Chip Move。但是一想那个是只更新某个点，这个是更新一段区间。其实一样，这个更新一段区间可以先更新差分，然后再求一个前缀和就可以了。
3. 也就是定义dp[i]为走到i的概率。先枚举步数，因为步数肯定是递增的，然后枚举每个点会对下一层做的贡献，那么就是对下一层的一段区间做贡献，先维护个差分，处理完之后。对差分求前缀和那么现在就是下一层到达每个点的概率。那么再复制给dp就行了。那么很明显就是一个滚动数组，因为这一层的只会用上一层的。答案加的时候就在每一层到达n号点的概率都求个平方就可以了。

代码：

#include<bits/stdc++.h>
#define int long long
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;

const int N = 8e3+10,mod = 998244353;

int n;
int va[N];
int dp[N],tp[N];
int sum[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;
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		cin>>va[i];
		sum[i] = ksm(va[i],mod-2)%mod; //记得预处理下要不然就超时了。
	}
	dp[1] = 1;
	int ans = 0;
	for(int j=1;j<=n-1;j++)
	{
		for(int i=0;i<=n;i++) tp[i] = 0;
		for(int i=1;i<n;i++)
		{
			int l = i+1,r = i+va[i];
			tp[l] = (tp[l]+dp[i]*sum[i]%mod)%mod;
			tp[r+1] = (tp[r+1]-dp[i]*sum[i]%mod)%mod;
		}
		for(int i=1;i<=n;i++) tp[i] = (tp[i]+tp[i-1])%mod;
		for(int i=1;i<=n;i++) dp[i] = tp[i];
		ans = (ans+dp[n]*dp[n]%mod)%mod; //走了j步到达n号点的概率
	}
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}

总结：
多多思考，类比推理。