DP
此题可以分为两个子问题。
(m)等于(1):
原题目转化为求一行数列里的(k)块区间的和,区间可以为空的值。
直接定义状态(dp[i][t])表示前i个数分为t块的最大值。
因为区间可以为空,所以最大值再小也不会比0小,所以初始化(dp)值为(0)。
有方程(dp[i][t]=max(dp[i-1][t],dp[j][t-1]+sum_{k=j+1}^{i}a[k]))
考虑顺序及边界,发现此时的(dp[i][t])是从(t-1)块转移过来的,所以(j)要和(i)断开。但是又因为可以从(1)一路取到(i),因此(j+1)可以取到(1)。所以(0<=j<i),而且(t)要放在循环的最外面。
(m)等于(2):
同样可以(dp),定义状态(dp[i][j][t])为第一列取到(i),第二列取到(j),分为(t)块的最大值。
有方程(dp[i][j][t]=max(dp[i-1][j][t],dp[i][j-1][t],dp[o][j][t-1]+sum_{k=o+1}^{i}a[k][1],dp[i][o][t-1]+sum_{k=o+1}^{j}a[k][2]);)
如果i等于j的话,还有方程
(dp[i][j][t]=max(dp[i][j][t],dp[o][o][t-1]+sum_{k=o+1}^ia[k][1]+a[k][2]);)
顺序及边界和第一问一样。
#include <bits/stdc++.h>
#define N 1001
using namespace std;
int n, m, k, maxn;
int dp[N][N][40], a[N][N], sum1[N], sum2[N], dp1[N][40];
int Max(int a, int b, int c, int d) {return max(max(a, b), max(c, d));}
int Q1(int i, int j)
{
return sum1[j] - sum1[i - 1];
}
int Q2(int i, int j)
{
return sum2[j] - sum2[i - 1];
}
int main()
{
scanf("%d%d%d", &n, &m, &k);
if (m == 1)
{
for (int i = 1; i <= n; i++)
scanf("%d", &a[i][1]), sum1[i] = sum1[i - 1] + a[i][1];
for (int t = 1; t <= k; t++)
for (int i = 1; i <= n; i++)
{
dp1[i][t] = dp1[i - 1][t];
for (int j = 0; j < i; j++)
dp1[i][t] = max(dp1[i][t], dp1[j][t - 1] + Q1(j + 1, i));
}
printf("%d", dp1[n][k]);
}
else
{
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
scanf("%d", &a[i][j]);
for (int i = 1; i <= n; i++)
sum1[i] = sum1[i - 1] + a[i][1],
sum2[i] = sum2[i - 1] + a[i][2];
for (int i = 1; i <= n; i++)
dp[i][0][1] = sum1[i], dp[0][i][1] = sum2[i];
for (int t = 1; t <= k; t++)
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
{
dp[i][j][t] = max(dp[i - 1][j][t], dp[i][j - 1][t]);
for (int o = 0; o < i; o++)//选择到第i位。
dp[i][j][t] = max(dp[i][j][t], dp[o][j][t - 1] + Q1(o + 1, i));
for (int o = 0; o < j; o++)
dp[i][j][t] = max(dp[i][j][t], dp[i][o][t - 1] + Q2(o + 1, j));
if (i == j)
for (int o = 0; o < i; o++)
dp[i][j][t] = max(dp[i][j][t], dp[o][o][t - 1] + Q1(o + 1, i) + Q2(o + 1, j));
}
printf("%d", dp[n][n][k]);
}
return 0;
}