【模版】m子段和的最大值

【模版】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])
优化时间复杂度到n²

测试题目

Max Sum Plus Plus

测试样例

输入：
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;
}