2022牛客多校9 BTwo Frogs（概率DP）

 

示例1

输入

5
1 1 1 1

输出

1

示例2

输入

5
4 3 2 1

输出

440198031

题目大意：这里有n个荷叶，前n-1个荷叶上都有一个数，第i个荷叶上的数字a[i]表示当位于该荷叶上时，下一步能跳到的范围是( i,i+a[i] ]，有两只青蛙，问我们这两只青蛙从1号跳到n号荷叶的步数相同的概率是多少。

解题思路：

状态表示：dp[i][j]表示从第i个点跳j步到达第n个点的概率。

状态转移：dp[i][k]=从( i,i+a[i] ]区间内的任意一点跳k-1步到达n点的概率之和，这里为了能快速求出一段区间的概率之和我们采用了前缀和的思想来进行优化，即s数组，其含义在代码中有注释。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const long long mod=998244353;
const int N=8010;
int n,a[N],dp[N][N],s[N][N],inv[N];
int qpow(int a,int b)
{
	int ans=1;
	while(b)
	{
		if(b&1) ans=1ll*ans*a%mod;
		a=1ll*a*a%mod;
		b>>=1;
	}
	return ans;
}
int main()
{	
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	scanf("%d",&a[i]);
	inv[0]=1;
	for(int i=n;i>0;i--)	 
	inv[i]=qpow(i,mod-2);
	dp[n][0]=1;
	s[0][n]=1;
	for(int k=1;k<=n;k++)
	{
		for(int i=1;i<=n;i++)
		s[k-1][i]=(s[k-1][i]+s[k-1][i-1])%mod;//s[k-1][i]记录的是从i以及之前的点跳k-1步到达n点的概率之和
		for(int i=n-1;i>=1;i--)
		{
			dp[i][k]=(dp[i][k]+(1ll*s[k-1][i+a[i]]-s[k-1][i])%mod*inv[a[i]]%mod)%mod;
			s[k][i]=dp[i][k];
		} 
	}
	long long ans=0;
	for(int i=0;i<=n;i++)
	ans=(ans+1ll*dp[1][i]*dp[1][i]%mod)%mod;
	printf("%lld\n",ans);
	return 0;
}