[BZOJ 1084] [SCOI2005] 最大子矩阵 【DP】

题目链接：BZOJ - 1084

题目分析

我看的是神犇BLADEVIL的题解。

1）对于 m = 1 的情况， 首先可能不取 Map[i][1]，先 f[i][k] = f[i - 1][k]; 再考虑取一段新的的情况，用 max(f[j][k - 1] + Sum[i][1] - Sum[j][1]) (0 <= j < i) 更新 f[i][j]；

2) 对于 m = 2 的情况，用 f[i][j][k] 表示左列取到第 i 个，右列取到第 j 个，共 k 个矩形的最优值。

首先还是可能不取新的矩形，那么 f[i][j][k] = max(f[i - 1][j][k], f[i][j - 1][k]);

之后可能左列取一个新的矩形，用 max(f[ii][j][k - 1] + Sum[i][1] - Sum[ii][1]) (0 <= ii < i) 更新 f[i][j][k];

可能在右列取一个新的矩形，用 max(f[i][jj][k - 1] + Sum[j][2] - Sum[jj][2]) (0 <= jj < j) 更新 f[i][j][k];

若 i == j, 那么可能取一个跨两列的矩形，用 max(f[ii][ii][k - 1] + Sum[i][1] + Sum[i][2] - Sum[ii][1] - Sum[ii][2]) (0 <= ii < i) 更新 f[i][j][k];

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>

using namespace std;

const int MaxN = 100 + 5, MaxK = 10 + 5;

int n, m, EK;
int Map[MaxN][3], Sum[MaxN][3], f1[MaxN][MaxK], f2[MaxN][MaxN][MaxK];

inline int gmax(int a, int b) {return a > b ? a : b;}
inline int gmin(int a, int b) {return a < b ? a : b;}

int main()
{
scanf("%d%d%d", &n, &m, &EK);
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= m; ++j) {
scanf("%d", &Map[i][j]);
Sum[i][j] = Sum[i - 1][j] + Map[i][j];
}
}
if (m == 1) {
for (int i = 1; i <= n; ++i) {
for (int k = 1; k <= EK; ++k) {
f1[i][k] = f1[i - 1][k];
for (int j = 0; j < i; ++j) {
f1[i][k] = gmax(f1[i][k], f1[j][k - 1] + Sum[i][1] - Sum[j][1]);
}
}
}
printf("%d\n", f1
[EK]);
}
else {
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= n; ++j) {
for (int k = 1; k <= EK; ++k) {
f2[i][j][k] = gmax(f2[i - 1][j][k], f2[i][j - 1][k]);
for (int jj = 0; jj < i; ++jj)
f2[i][j][k] = gmax(f2[i][j][k], f2[jj][j][k - 1] + Sum[i][1] - Sum[jj][1]);
for (int jj = 0; jj < j; ++jj)
f2[i][j][k] = gmax(f2[i][j][k], f2[i][jj][k - 1] + Sum[j][2] - Sum[jj][2]);
if (i == j) {
for (int jj = 0; jj < i; ++jj)
f2[i][j][k] = gmax(f2[i][j][k], f2[jj][jj][k - 1] + Sum[i][1] + Sum[i][2] - Sum[jj][1] - Sum[jj][2]);
}
}
}
}
printf("%d\n", f2

[EK]);
}
return 0;
}