【模版】m子段和的最大值
写在前面
题意:给定一个长度为n的序列,求出m个子段序列的最大和.
参考网站:
https://blog.csdn.net/codeswarrior/article/details/80310401
https://www.cnblogs.com/wuwangchuxin0924/p/6546901.html
解题思路
状态dp[i][j]
表示前j
个数分成i
组的最大值
动态转移方程:dp[i][j]=max(dp[i][j-1]+a[j],max(dp[i-1][k])+a[j]) (0<k<j)
即比较两种情况:
1.第a[j]
元素单独作为第i
个字段
2.第a[j]
元素和前面的字段共同当做第i
个字段
在每次计算dp[i][j]
的时候记录下0~j-1
的最大值,用数组Max[]
保存下来
dp[i][j]
成为dp[j]
,新的状态转移方程:dp[j] = max(dp[j-1]+a[j], Max[j-1]+a[j])
优化时间复杂度到n2
测试题目
测试样例
输入:
1 3 1 2 3
2 6 -1 4 -2 3 -2 3
输出:
6
8
代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define inf 0x3f3f3f3f
using namespace std;
const int MaxSize = 1e6+10;
#define mm(a) memset(a, 0, sizeof(a))
int a[MaxSize]; // 存放源数组
// dp[j-1]存放前j-1个数分成i个组得到的最大总和
// j不单独成组, 因此j必须放在前j-1个数的最后一组里面
int dp[MaxSize];
// Max[j-1]存放1~j-1个数分成i-1组取得的最大总和
// j单独成组, 自身成为第i组
int Max[MaxSize];
int main() {
// n个数 m组 mmax最大总和
int n, m, mmax = -inf;
while (~scanf("%d%d", &m, &n)){
for (int i = 1; i <= n; i++) scanf("%d", &a[i]); // 获得源数组
mm(dp);
mm(Max);
for (int i = 1; i <= m; i++){ // 分成i个组
// 开始新的组数分配时初始化无穷小
mmax = -inf;
// 前j个数分成i组, 至少需要i个数
for (int j = i; j <= n; j++){
dp[j] = max(dp[j-1]+a[j], Max[j-1]+a[j]);
// 比较两种情况(第j个数是否单独成组) 取最大值
// mmax是上一次循环得到的, 因此更新的是j-1
Max[j-1] = mmax;
mmax = max(mmax, dp[j]);
}
}
printf("%d\n", mmax); // 输出得到的最大总和
}
return 0;
}