基准时间限制:2 秒 空间限制:131072 KB 分值: 40
难度:4级算法题
一个M*N的矩阵,找到此矩阵的一个子矩阵,并且这个子矩阵的元素的和是最大的,输出这个最大的值。
例如:3*3的矩阵:
-1 3 -1
2 -1 3
-3 1 2
和最大的子矩阵是:
3 -1
-1 3
1 2
Input
第1行:M和N,中间用空格隔开(2 <= M,N <= 500)。 第2 - N + 1行:矩阵中的元素,每行M个数,中间用空格隔开。(-10^9 <= M[i] <= 10^9)
Output
输出和的最大值。如果所有数都是负数,就输出0。
Input示例
3 3 -1 3 -1 2 -1 3 -3 1 2
Output示例
7
题解:对于一个矩形,如果左右边界确定,那么我们可以转变为求左右边界下的最大子段和,最大子段和O(n),枚举左右边界O(n ^ 2),故时间复杂度为O(n ^ 3);
AC代码
#include <iostream>
#include <vector>
#include <stack>
#include <climits>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstring>
typedef long long ll;
using namespace std;
const ll maxn = 555;
ll p[maxn][maxn];
int main(){
ll n, m, t;
scanf("%lld%lld", &m, &n);
for(ll i = 0; i < n; i++){
for(ll j = 0; j < m; j++){
scanf("%lld", &t);
p[i][j] = (j == 0 ? t : p[i][j - 1] + t);
}
}
ll ans = 0;
for(ll i = 0; i < m; i++){
for(ll j = i; j < m; j++){
ll sum = 0;
for(ll k = 0; k < n; k++){
sum += p[k][j] - (i == 0 ? 0 : p[k][i - 1]);
if(sum < 0)
sum = 0;
ans = max(ans, sum);
}
}
}
printf("%lld\n", ans);
return 0;
}