历届试题 最大子阵

题目链接 

问题描述

　　给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。
　　其中，A的子矩阵指在A中行和列均连续的一块。

分析：最大子阵和，可以将几行数据合成一行，求出合并后单行数据的最大子序列的和，枚举全部的合并列即可。

#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 argc, char** argv) {
	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;
}

更好的解法：用了前缀和减少了每次合并的计算。

原文链接：https://blog.csdn.net/update7/article/details/56681524 

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

输入格式

　　输入的第一行包含两个整数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。