CF1093F

fi,j 表示前 i 个位置,第 i 个位置的数字是 j 的方案数,si=jfi,jmxi,j 为位置 i 往前全是 j 的最长长度。

fi,j={si1                                      mxi,j<lensi1(silenfilen,j)     mxi,jlen

上面显然,因为随便选一定合法。

si1(silenfilen,j) 的意义是,之前的随便选的方案数 si1,减去这个位置恰好出现了 len 个连续的方案数 silenfilen,j,由于新加一个数只可能出现恰好 len 个的非法情况,所以这么减刚好就是真实的答案。

时间复杂度 O(nk)

Code:

#include <bits/stdc++.h>
using namespace std;
const int N = 100005, M = 105, mod = 998244353;
int n, k, len;
int a[N], f[N][M], s[N], mx[N][M];

int main() {
	scanf("%d%d%d", &n, &k, &len);
	for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= k; ++j)
			if (a[i] == j || a[i] == -1) mx[i][j] = mx[i - 1][j] + 1;
	s[0] = 1;
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= k; ++j) {
			if (a[i] == j || a[i] == -1) {
				f[i][j] = s[i - 1];
				if (mx[i][j] >= len) {
					int tmp = (s[i - len] - f[i - len][j] + mod) % mod;
					f[i][j] = (f[i][j] - tmp + mod) % mod;
				}
				s[i] = (s[i] + f[i][j]) % mod;
			}
		}
	printf("%d", s[n]);
	return 0;
}
posted @   Kobe303  阅读(21)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话
点击右上角即可分享
微信分享提示