最大子矩阵【蓝桥杯】

题目描述

问题描述
　　给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。
　　其中，A的子矩阵指在A中行和列均连续的一块。
输入格式
　　输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
　　接下来n行，每行m个整数，表示矩阵A。
输出格式
　　输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
样例说明
　　取最后一列，和为10。
数据规模和约定
　　对于50%的数据，1<=n, m<=50；
　　对于100%的数据，1<=n, m<=500，A中每个元素的绝对值不超过5000。

涉及的知识点：最大字段和问题（一维数组） 和 二维数组的前缀和问题
解题思路：将二维数组按列求前缀和，那么 pre[j][k] - pre[i-1][k] 就是第k列从第i行加到第j行的数列和，把他当作一维数组求前缀和，即再按照行从左到右求最大值即可。

代码如下：

#include<iostream>
#include<vector>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
int v[505][505];
int pre[505][505];
int m,n;
int main() {
	memset(v,0,sizeof(v));
	memset(pre,0,sizeof(pre));
	scanf("%d%d",&m,&n);
	for(int i=1; i<=m; i++)
		for(int j=1; j<=n; j++) 
			scanf("%d",&v[i][j]);
	//求列向前缀和
	for(int i=1; i<=n; i++) 
		for(int j=1; j<=m; j++) 
			pre[j][i] = v[j][i] + pre[j-1][i];
	int ans=v[1][1];	
	for(int i=1; i<=m; i++) 
	{
		for(int j=i; j<=m; j++) 
		{
			int sum =0;
			for(int k=1; k<=n; k++) 
			{
				sum = max(sum,0) + pre[j][k]-pre[i-1][k];
				//如果是减去第i行，那么永远加不上第i行 
				ans = max(ans,sum);
			}
		}
	}
	printf("%d",ans);
	return 0;
}