蓝桥杯:最大子阵

问题描述

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

Thinking:

开始做的时候,只想到了n*n*m*m时间复杂度的算法,数据没能通过.之后参考了一些网上大神的代码,突然觉悟,其实就是最大子段和问题的扩展.

#include<iostream>
#define MIN -5001
using namespace std;
int main()
{
	bool hasUpdate=false;
	int i,j,k,row,col,temp,maxElem=MIN,tempSum,maxSum=0;
	cin>>row>>col;
	
	int **record=new int*[row+1];
	for(i=0;i<=row;i++)
		record[i]=new int[col+1]();
		
	for(i=1;i<=row;i++)
	{
		for(j=1;j<=col;j++)
		{
			cin>>temp;
			if(temp>maxElem)//find the minimum number of the whole array.
				maxElem=temp;
			record[i][j]=record[i-1][j]+temp;
		}
	}
	
	for(i=1;i<=row;i++)
	{
		for(j=i;j<=row;j++)
		{
			tempSum=0;
			for(k=1;k<=col;k++)//the maximum-sum sub-array algorithm
			{
				temp=record[j][k]-record[i-1][k];
				if(tempSum>0)
					tempSum+=temp;
				else
					tempSum=temp;
				if(tempSum>maxSum)
				{
					maxSum=tempSum;
					hasUpdate=true;
				}
			}
		}
	}
	
	if(hasUpdate)
		cout<<maxSum<<endl;
	else//if maxSum never update, the maxSum is maxElem.
		cout<<maxElem<<endl;
		
	return 0;
}