初见蓝桥——历年试题 最大子阵

  历届试题 最大子阵  

时间限制：1.0s   内存限制：256.0MB

问题描述

　　给定一个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。

这是一道动态规划的题目，思想与最大字段问题相同，只是在最初需要列举一些连续的行和列，代码如下：

#include<iostream>
#include<string.h>
#define MAXSIZE 505
#define INF 0x3f3f3f3f  
using namespace std;
int a[MAXSIZE][MAXSIZE];
int b[MAXSIZE];
int dp[MAXSIZE];
int ans=-INF;
int main()
{
	int N,M;
	cin>>N>>M;
	for(int i=0; i<N; i++)
	for(int j=0; j<M; j++)
	cin>>a[i][j];
	for(int sr=0; sr<N; sr++)
	{
		memset(b,0,sizeof(b));
		for(int i=sr; i<N; i++){
			for(int j=0; j<M; j++){
				b[j]+=a[i][j];
			}
			dp[0]=b[0];
			if(dp[0]>ans)
			ans=dp[0];
			for(int j=1; j<M; j++){
				if(dp[j-1]<0)
				dp[j]=b[j];
				else 
				dp[j] = dp[j-1]+b[j];
				if(dp[j]>ans)
				ans=dp[j];
			}
		}
	}
	cout<<ans;
	return 0; 
}