题目链接
我们N个数中去取M个互不相交的区间段,求最大的全体所要的区间段的和。
那么,我们怎样去维护这个DP呢?我们先不考虑内存(这题卡了内存倒是难受的很)。维护两个东西,一个dp[i][j]我们现在查到第i个物品,并且在分成j个块的时候,还强制要选择第i个物品的时候的最大区间段的和。
这时候的DP是这样的:
for(int i=1; i<=N; i++)
{
scanf("%d", &a[i]);
}
for(int j=1; j<=M; j++) //分j块时候的最优解
{
for(int i=j; i<=N; i++)
{
if(i == j) dp[i][j] = dp[i-1][j-1] + a[i];
else dp[i][j] = max(dp[i-1][j] + a[i], mx[i-1][j-1] + a[i]);
if(i == j) mx[i][j] = dp[i][j];
else mx[i][j] = max(mx[i-1][j], dp[i][j]);
}
}
for(int i=M; i<=N; i++) ans = max(ans, dp[i][M]);
在这里,我们发现dp[i][j]里的j是可以用滚动数组来代替的,因为处理的是分的区间段,而且在外层循环。但是就这样只有一层滚动数组的,我们开这样的空间仍然会MLE,那么我们还需要去把mx数组的大小也尽可能的优化一下。
之后,我们发现,mx[]数组向后递推的时候,只会受前一刻的影响,所以我们可以不妨去减少一维,那样子,就变成了我们最后想要的答案了。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
#define MP3(a, b, c) MP(MP(a, b), c)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1e6 + 7;
int M, N, a[maxN];
ll dp[maxN][2], mx[maxN], ans;
int main()
{
while(scanf("%d%d", &M, &N) != EOF)
{
ans = -INF;
for(int i=0; i<=N; i++) dp[i][0] = dp[i][1] = mx[i] = 0;
for(int i=1; i<=N; i++)
{
scanf("%d", &a[i]);
}
for(int j=1; j<=M; j++) //分j块时候的最优解
{
for(int i=j; i<=N; i++)
{
if(i == j) dp[i][j&1] = dp[i-1][(j-1)&1] + a[i];
else dp[i][j&1] = max(dp[i-1][j&1] + a[i], mx[i-1] + a[i]);
if(i - 1 == j) mx[i - 1] = dp[i - 1][j&1];
else mx[i - 1] = max(mx[i-2], dp[i - 1][j&1]);
}
}
for(int i=M; i<=N; i++) ans = max(ans, dp[i][M&1]);
printf("%lld\n", ans);
}
return 0;
}