问题 1445: [蓝桥杯][历届试题]最大子阵

问题 1445: [蓝桥杯][历届试题]最大子阵

时间限制: 1Sec 内存限制: 128MB 提交: 635 解决: 173

题目描述

给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。 

其中，A的子矩阵指在A中行和列均连续的一块。 

样例说明 
取最后一列，和为10。 
数据规模和约定 
对于100%的数据，1< =n,  m< =500，A中每个元素的绝对值不超过5000。 
 

输入

输入的第一行包含两个整数n,  m，分别表示矩阵A的行数和列数。 
接下来n行，每行m个整数，表示矩阵A。 

输出

输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。 

样例输入

3  3 
-1  -4  3 
3  4  -1 
-5  -2  8 

样例输出

10

#include <cstdio>
#include <cstring>
using namespace std;
const int N = 500+5, INF = 0x3f3f3f3f;
int matrix[N][N], d[N], sum[N];
int main() {
	int n, m;
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= m; j++)
			scanf("%d",&matrix[i][j]);
	int maxSum = -INF;
	for(int k = 1; k <= n; k++){
		memset(sum,0,sizeof(sum));
		for(int i = k; i <= n; i++){
			d[0] = 0;
			for(int j = 1; j <= m; j++){
				sum[j] += matrix[i][j];
				if( d[j-1] > 0)	d[j] = d[j-1]+sum[j];
				else  d[j] = sum[j];
				if(d[j] > maxSum) maxSum = d[j];
			}	
		}
	}
	printf("%d\n",maxSum);
	return 0;
}