“蔚来杯“2022牛客暑期多校训练营9-B Two Frogs

ren1xin

已于 2022-08-20 19:35:29 修改

阅读量251

点赞数

分类专栏：多校联赛 “蔚来杯“ 文章标签：算法 c++ 动态规划概率论

于 2022-08-20 19:30:49 首次发布

本文链接：https://blog.csdn.net/qq_60252920/article/details/126441613

版权

多校联赛同时被 2 个专栏收录

40 篇文章 0 订阅

订阅专栏

“蔚来杯“

9 篇文章 0 订阅

订阅专栏

"蔚来杯"2022牛客暑期多校训练营9-B Two Frogs

原题题面：https://ac.nowcoder.com/acm/contest/33194/B

文章目录

"蔚来杯"2022牛客暑期多校训练营9-B Two Frogs

题目大意

有 $n(1\le n\le 8000)$ 个荷叶排成一排，两只青蛙需要从第 $1$ 个荷叶跳到第 $n$ 个荷叶上。如果青蛙在第 $i (i < n)$ 个荷叶上，则其可以跳跃到 $i,i+a_i]$ 区间的荷叶上。

两只青蛙每次跳跃会等概率随机选择一个合法的荷叶目的地，求他们以相同的跳跃次数跳跃到第 $n$ 个荷叶上的概率。

解题思路

设 $dp_{i,j}$ 表示调到第 $i$ 个荷叶需要 $j$ 次的概率。 $dp_{i,j}=\sum_{k|k+a_k\le i}dp_{k,j-1}$ 。由于要枚举 $i, j, k$ 所需复杂度 $O(n^3)$ ，超时。

发现 $dp_{i,j}$ 至 $dp_{i+a_i,j}$ 所得贡献相同，可以先差分标记贡献，再在用前求出前缀和，可将复杂度降到 $O(n^2)$ 。

代码实现

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=8005,mod=998244353;
int n,f[N][N],a[N],ans,fp[N];
int fpow(int x,int p){
	int ret=1;
	for(;p;x=(ll)x*x%mod,p>>=1)if(p&1)ret=(ll)ret*x%mod;
	return ret;
}
int main(){
	cin>>n;
	for(int i=1;i<n;i++)cin>>a[i],fp[i]=fpow(a[i],mod-2);
	f[1][0]=1,f[2][0]=-1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<i;j++)f[i][j]=(f[i][j]+f[i-1][j])%mod;
		for(int j=0;j<i;j++){
			f[i+1][j+1]=((ll)f[i][j]*fp[i]%mod+f[i+1][j+1])%mod;
			f[i+a[i]+1][j+1]=(f[i+a[i]+1][j+1]-(ll)f[i][j]*fp[i]%mod+mod)%mod;
		}
	}
	for(int i=0;i<n;i++)ans=((ll)ans+(ll)f[n][i]*f[n][i]%mod)%mod;
	cout<<ans;
}