题意
把一个长度为n的数组分成m段,使得
sum(i1, j1) + sum(i2, j2) + ... + sum(im, jm) 最大
题解
[线性DP -- 子段和模型]
6 6 13 5 -14 -5 -13 19
状态表示:
dp[i][j]: 共分为i段,且选择a[j]的方案集合
属性:
Max
集合划分:
对于当前a[j]:
1. 和前面共成一段
dp[i][j] = dp[i][j-1] + a[j]
2. 自成一段
dp[i][j] = dp[i-1][k] + a[j], 1<=k<j
时间复杂度 O(m*n*n)
空间复杂度 O(m * n)
n<=1e6, 肯定过不了
优化:
刷表,发现
(i, j) 只用到 (i, j-1) 和 max{(i-1, k)} 1<=k<j
故而想到用:
用一个额外的数组记录 Max{(i-1, k)}
令 Max[i][j] = max{dp[i][1...j]}
故而 Max[i][j] = max(Max[i][j-1], dp[i][j])
进而:
dp[i][j] = max(dp[i][j-1] + a[j], Max[i-1][j-1] + a[j])
下一个问题:
数组开多大? n <= 1e6, m并没有给
所以考虑用滚动数组优化
dp[i][j] 只用到 dp[i][j-1], 所以 dp[]用一维
Max[i][j] 用到 Max[i-1][j], 所以 Max[] 用二维
初始状态:
i < j, 不可达状态 -INF
i == j, dp[i][j] = dp[i-1][j-1] + a[j];
故而, dp[0][0] = 0
Max[0][i] = 0, 其他的 Max[1...m][i] = -INF
目标状态:
max(dp[m][i]) (m<=i<=n)
Max[m][n]
// 根据定义:Max[m][n] = max({dp[m][1...n]})
Code
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
#define read(a) scanf("%d", &a)
#define readl(a) scanf("%lld", &a)
#define readf(a) scanf("%lf", &a)
#define readc(a) scanf("%c", &a)
#define reads(a) scanf("%s", a)
#define Buff ios::sync_with_stdio(false)
#define mem(a) memset(a, 0, sizeof a)
#define pb push_back
const ll INF = 1e16 + 7;
const int N = 1e6 + 7;
const int M = 1e6 + 7;
const ll mod = 1e9 + 7;
ll dp[N], Max[2][N], a[N];
signed main()
{
int n, m;
while (~scanf("%d%d", &m, &n))
{
for(int i = 1; i <= n; i++) readl(a[i]);
memset(dp, -0x3f, sizeof dp);
memset(Max[0], 0, sizeof Max[0]);
dp[0] = 0;
for(int i = 1; i <= m; i++)
{
Max[i&1][i-1] = -INF;
for(int j = i; j <= n; j++)
{
dp[j] = max(Max[i-1&1][j-1], dp[j-1]) + a[j];
Max[i&1][j] = max(Max[i&1][j-1], dp[j]);
}
}
ll ans = -INF;
for(int i = m; i <= n; i++) ans = max(ans, dp[i]);
}
return 0;
}