蒜头君的最大子矩阵和

题目:
蒜头君的最大子矩阵和
代码如下:

#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行的和,之后,我们先要把行的上界和下界固定下来,然和把每一列的数都加起来。想象一下其实就是相当于把一个矩阵压缩成一列数,然后用求最大子段和的方法求出最大值。随后,我们需要尝试每一种可能,也就是每一种上下界不同的情况下的最大矩阵和,找出最大的输出即可。

这道题刚开始想到用暴力枚举去做,但这里的数据明显是过不去的,最后还是看了题解才明白的…