CF1093F
设 fi,j 表示前 i 个位置,第 i 个位置的数字是 j 的方案数,si=∑jfi,j,mxi,j 为位置 i 往前全是 j 的最长长度。
fi,j={si−1 mxi,j<lensi−1−(si−len−fi−len,j) mxi,j≥len
上面显然,因为随便选一定合法。
而 si−1−(si−len−fi−len,j) 的意义是,之前的随便选的方案数 si−1,减去这个位置恰好出现了 len 个连续的方案数 si−len−fi−len,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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话