题目地址:
https://www.acwing.com/problem/content/description/128/
给定一个包含整数的二维矩阵,子矩形是位于整个阵列内的任何大小为 1 × 1 1×1 1×1或更大的连续子阵列。矩形的总和是该矩形中所有元素的总和。在这个问题中,具有最大和的子矩形被称为最大子矩形。例如,下列数组:
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
其最大子矩形为:
9 2
-4 1
-1 8
它拥有最大和 15 15 15。
输入格式:
输入中将包含一个
N
×
N
N×N
N×N的整数数组。第一行只输入一个整数
N
N
N,表示方形二维数组的大小。
从第二行开始,输入由空格和换行符隔开的
N
2
N^2
N2个整数,它们即为二维数组中的
N
2
N^2
N2个元素,输入顺序从二维数组的第一行开始向下逐行输入,同一行数据从左向右逐个输入。
数组中的数字会保持在
[
−
127
,
127
]
[−127,127]
[−127,127]的范围内。
输出格式:
输出一个整数,代表最大子矩形的总和。
数据范围:
1
≤
N
≤
100
1≤N≤100
1≤N≤100
枚举上下边界,考虑所有上下边界确定的子矩阵,接着问题变为考虑一维情况下的最大子数组的和,这可以用动态规划来做,枚举右端点,然后按照子数组是单独一个数,还是加上前面的来分类。代码如下:
#include <iostream>
using namespace std;
const int N = 110;
int n;
int a[N][N], s[N][N];
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++) {
scanf("%d", &a[i][j]);
s[i][j] = s[i - 1][j] + a[i][j];
}
int res = -2e9;
for (int i = 1; i <=n; i++)
for (int j = i; j <= n; j++) {
int cur = 0;
for (int k = 1; k <= n; k++) {
cur = max(cur, 0) + s[j][k] - s[i - 1][k];
res = max(res, cur);
}
}
printf("%d\n", res);
}
时间复杂度 O ( N 3 ) O(N^3) O(N3),空间 O ( N 2 ) O(N^2) O(N2)。