[abc] abc 253E - Distance Sequence

前言

$tag:$ 计数 前缀和优化dp 难
传送门 :

题意 :
给定$N,M$询问有多少种序列满足条件

条件如下 :
$\forall i,(1\le a[i]\le M)$
$|a|=N$
$|a_i-a_{i+1}|\ge K$
即长度$N$,值域是$M$,相邻元素的差绝对值大于$K$

思路 :
状态表示 :
$dp[i][j]$ 前$i$个并且以$j$结尾的方案数

状态转移 :
$dp[i+1][j]=(dp[i][1]+....+dp[i][j-K])+(dp[i][j+K]+....dp[i][M])$

时间复杂度$O(N{M}^2)$显然$TLE$

但是其实我们并不需要去枚举转移

我们发现,我们可以单独计算出来上一层的$dp[i][1]+....dp[i][j-k]$ 这种状态

显然这里可以使用到 前缀和进行优化

因此时间复杂度将为$O(NM)$

code :

ll f[N][N];
ll s[N];
int n,m,k;

void solve(){
	cin>>n>>m>>k;
	
	
	for(int i=1;i<=m;i++) f[1][i] = 1;
	
	for(int i=2;i<=n;i++){
		for(int j=1;j<=m;j++)
		s[j] = (s[j-1]+f[i-1][j])%mod;
		
		for(int j=1;j<=m;j++){
			int l =  j-k ;
			int r =  j+k;
			
			if(k == 0) f[i][j] = (f[i][j]+s[m])%mod;
			else{
				if(l>0) f[i][j] = (f[i][j]+s[l])%mod;
				if(r<=m) f[i][j] = (f[i][j]+s[m] - s[r-1]+mod)%mod;
			}
		}
	}
	ll ans = 0;
	for(int i=1;i<=m;i++)
	ans=(ans+f[n][i])%mod;
	cout<<ans<<endl;
	
}