题目大意:给定m和n,再给出n个数,求出m段数的和使总和最大
解法1:设f[i][j][k]表示状态,记录最大值
f[i][j][0]表示当前是第i个数,已经选取了j段,第i个数没有被选
f[i][j][1]表示当前是第i个数,已经选取了j段,第i个数被选
dp[i][j][0] = max(dp[i - 1][j][0], dp[i - 1][j][1]);
dp[i][j][1] = max(dp[i - 1][j - 1][0], max(dp[i - 1][j][1], dp[i - 1][j - 1][1])) + a[i];
这样的话会超出空间限制,因此用滚动数组优化第一维空间。
代码如下
#include <iostream>
#include <cstring>
using namespace std;
const int N = 1e6 + 10;
int dp[2][N][2];
int a[N];
int n, k;
int main()
{
while (scanf("%d%d", &k, &n) != -1)
{
for (int i = 1; i <= n; i ++)
scanf("%d", &a[i]);
memset(dp, -0x3f, sizeof dp);
dp[0][0][0] = 0;
dp[1][0][0] = 0;
for (int i = 1; i <= n; i ++)
for (int j = 1; j <= k; j ++)
{
dp[i & 1][j][0] = max(dp[(i - 1) & 1][j][0], dp[(i - 1) & 1][j][1]);
dp[i & 1][j][1] = max(dp[(i - 1) & 1][j - 1][0], max(dp[(i - 1) & 1][j][1], dp[(i - 1) & 1][j - 1][1])) + a[i];
}
cout << max(dp[n & 1][k][1], dp[n & 1][k][0]) << endl;
}
}
第二种解法
dp[i][j]表示已经选取第j个数,选取的段数为i的最大值,网上搜索的前几个都是这个解法