蒜头君的最大子矩阵和
题目:
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define NIL 1e9
long long a[405][405],pre[405][405];
int main()
{
long long n,m,sum,ans = -NIL;
cin >> n >> m;
for(int i = 1;i <= n;i++){
for(int j = 1;j <= m;j++){
cin >> a[i][j];
ans = max(ans,a[i][j]);
}
}
if(ans < 0) cout << ans << endl; //矩阵所有值都是负数,相加只会越加越小,所以直接输出最大的那个
else{
for(int i = 1;i <= n;i++)
for(int j = 1;j <= m;j++)
pre[i][j] = pre[i - 1][j] + a[i][j]; //pre[i][j]代表第j列前i行总和
for(int i = 1;i <= n;i++){
for(int j = i;j <= n;j++){
sum = 0; //没选定上界和下界后总和清零
for(int k = 1;k <= m;k++){
if(sum + pre[j][k] - pre[i][k] < 0) sum = 0;
else sum += pre[j][k] - pre[i][k];
ans = max(ans,sum);
}
}
}
cout << ans << endl;
}
return 0;
}
这道题要求求最大子矩阵之和,这道题其实和最大子段和很像。首先我们用前缀和求出每一列上前i行的和,之后,我们先要把行的上界和下界固定下来,然和把每一列的数都加起来。想象一下其实就是相当于把一个矩阵压缩成一列数,然后用求最大子段和的方法求出最大值。随后,我们需要尝试每一种可能,也就是每一种上下界不同的情况下的最大矩阵和,找出最大的输出即可。
这道题刚开始想到用暴力枚举去做,但这里的数据明显是过不去的,最后还是看了题解才明白的…