面试经典(6)--连续子数组最大和--二维

前面写了一篇一维数组求连续子数组最大和http://blog.csdn.net/getnextwindow/article/details/23958529。二维的情况会复杂非常多会巧妙的借用一维的算法。

假设暴力解法,复杂度明显不尽如人意,由于求和的过程中,有非常多冗余情况,所以我们能够考虑将前面求和的结果保存下来,以空间换时间,这样能够减少复杂的。设PS[i][j]是以(i,j),(1,j),(i,1),(1,1)为顶点的矩形区域的元素之和。

略微分析一下得,PS[i][j]=PS[i-1][j]+PS[i][j-1]-PS[i-1][j-1]+A[i][j];

利用两层循环就能够递推得PS[i][j]。

我们前面提到,一维数组的连续子数组求和线性完毕,我们考虑是否能把二维的转换为一维。

上图就说明了问题,我们把a行c行之间的列看成一个“元素”,这样就转换成一维的情况。仅仅要枚举a、c就能够扫描出全部矩形。

代码例如以下:

#define MAX 100
int A[MAX][MAX];
long long PS[MAX][MAX];


long long MatrixSum(int a,int c,int i)
{
	return PS[c][i]-PS[c][i-1]-PS[a-1][i]+PS[a-1][i-1];
}

int find_max(int n,int m)
{
	int a,c,i,j;
	for(i=1;i<=m;i++)
		PS[0][i]=0;
	for(j=1;j<=n;j++)
		PS[j][0]=0;
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			PS[i][j]=PS[i-1][j]+PS[i][j-1]-PS[i-1][j-1]+M[i][j];
	int maxval=INT_MIN; 
	for(a=1;a<=n;a++)
	{
		for(c=a;c<=n;c++)
		{
			long long curSum=MatrixSum(a,c,1);
			long long maxsum=INT_MIN;
			for(i=1;i<=m;i++)
			{
				if(curSum<=0)
					curSum=MatrixSum(a,c,i);
				else
					curSum+=MatrixSum(a,c,i);
				if(curSum>maxsum)
					maxsum=curSum;
			}
			if(maxsum>maxval)
				maxval=maxsum;
		}
	}
	return maxval;
}

參考:http://blog.csdn.net/linyunzju/article/details/7723730

posted @ 2014-05-24 11:07  mfrbuaa  阅读(234)  评论(0编辑  收藏  举报