题意
输入n个数。选取m个不重叠的区间,使得m个区间内的数之和最大。
解题
要记录两个量:选取了的数的个数、选取的区间个数。故设dp[i][j]表示选取了i个不重叠区间、选取了j个数、最后一个区间包含第j个数的选取方式的值。
这样,dp[i][j]=max{dp[i][j-1],dp[i-1][k]}+s[j],k取值范围是[i-1,j-1]。
空间复杂度为O(n*m),时间复杂度为O(n*m*n)。
因为第i行的信息更新只需要用到第i-1行的信息,故可以通过滚动数组来优化空间,将二维数组降成一维。空间复杂度降为O(n)。
对于上式中dp[i-1][k],k取值范围是[i-1,j-1]。
设t[i][j]表示第i行第i-1列的值到第j-1列的值的最大值。同样用滚动数组维护,即设tmp[j]表示当前行[i-1,j-1]的最大值。
这样,通过dp预处理出dp[i-1][k]。时间复杂度降为O(n*m)。
AC代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <iostream>
#include <cmath>
#define INF 0x3f3f3f3f
using namespace std;
#define MAX 1000010
int w[2][MAX];
int dp[MAX];
int a[MAX];
int sum[MAX];
int main()
{
int m, n;
while (scanf("%d%d", &m, &n) > 0)
{
sum[0] = 0;
for (int i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
sum[i] = sum[i-1] + a[i];
w[0][i] = 0;
}
int t = 1;
for (int i = 1; i <= m; ++i)
{
w[t][i] = dp[i] = sum[i];
for (int j = i+1; j <= n-m+i; ++j)
{
dp[j] = max(dp[j-1], w[1-t][j-1]) + a[j];
w[t][j] = max(dp[j], w[t][j-1]);
}
t = 1 - t;
}
printf("%d\n", w[m%2][n]); //由于w退化了,没了w[m][n]因此m%2可以表示第m次,你可以用m = 1,和2试试便知道了
}
return 0;
}