CF958C3
首先有 O(n2k) 的暴力 DP。
设 fi,j 表示前 i 个数分成 j 段的最小和,枚举转移点 k:fi,j=min{fk,j−1+(si−sk)modp},其中 s 表示 a 的前缀和。
这时要注意到关键性质:fi,j≡si(modp)。
证明的话数学归纳一下就好了,大抵就是 (a+b)modp=((amodp)+(bmodp))modp。
那么对于 fi,j 的两个转移点 x,y(x≠y,x,y<i),设 wx=(si−sx)modp,wy=(si−sy)modp。
那么就是要在 fx,j−1+wx 和 fy,j−1+wy 中选最小。
假设 fx,j−1≤fy,j−1,那么由于 fx,j−1+wx≡fy,j−1+wy(modp),wx,wy<p,所以 fx,j−1+wx 一定不大于 fy,j−1+wy。
所以对于每个 j 记录一下最小的 fi,j 的位置 i 即可,从这个 i 转移来一定是最优的。
时间复杂度 O(nk)。
Code:
#include <bits/stdc++.h>
using namespace std;
const int N = 500005, M = 105;
int n, k, p;
int a[N];
int f[N][M], g[2][M];
void chkmin(int &a, int b) {
if (a > b) a = b;
}
int main() {
scanf("%d%d%d", &n, &k, &p); for (int i = 1; i <= n; ++i) scanf("%d", &a[i]), a[i] %= p, a[i] = (a[i] + a[i - 1]) % p;
memset(f, 0x3f, sizeof f), f[0][0] = 0;
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= k; ++j)
f[i][j] = min(f[i][j], f[g[(i - 1) & 1][j - 1]][j - 1] + (a[i] - a[g[(i - 1) & 1][j - 1]] + p) % p);
for (int j = 0; j <= k; ++j) {
g[i & 1][j] = g[(i - 1) & 1][j];
if (f[i][j] < f[g[i & 1][j]][j]) g[i & 1][j] = i;
}
}
printf("%d", f[n][k]);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话