二维前缀和

 题目描述

给你一个m×n的整数矩阵，在上面找一个x×y的子矩阵，使子矩阵中所有元素的和最大。

题目输入 

输入数据的第一行为一个正整数T，表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y（0<m,n<1000 AND 0<x<=m AND 0<y<=n），表示给定的矩形有m行n列。接下来这个矩阵，有m行，每行有n个不大于1000的正整数。

题目输出 

对于每组数据，输出一个整数，表示子矩阵的最大和。

Sample 

代码

#include<stdio.h>
int a[1010][1010],b[1010][1010];
int main()
{
	int N;
	int m,n,x,y;
	scanf("%d\n",&N);
	while(N--)
	{
	scanf("%d %d %d %d",&m,&n,&x,&y);
	for(int i=1;i<=m;i++)
		for(int j=1;j<=n;j++)
		scanf("%d",&a[i][j]);
		
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
		{
			b[i][j]=a[i][j]+b[i][j-1]+b[i-1][j]-b[i-1][j-1];
		}
	}	
	int max=0;
	for(int i=x;i<m;i++)
	{
		for(int j=y;j<n;j++)
		{			
	if(max<b[i][j]-b[i][j-x]-b[i-y][j]+b[i-x][j-y])
		max=b[i][j]-b[i][j-x]-b[i-y][j]+b[i-x][j-y];
		}
	}
	printf("%d",max);
}
	return 0;
 } 

已知给出的m×n的矩阵，求x×y的子矩阵的所有元素的和

假设二维前缀和为b[i][j],即从a[1][1]到a[i][j]构成矩阵的和(如图1所示)

图1:                                                              

 

要想求子矩阵的所有元素的和,则需要求出m×n矩阵的二维前缀和,由图2可知

从a[1][1]到a[i][j]构成矩阵的和可以看成“一点+两矩阵-多加的小矩阵”,注意:边缘部分

推得: b[i][j]=a[i][j]+b[i][j-1]+b[i-1][j]-b[i-1][j-1];

   图2:

 

得到矩阵的前缀和公式后,便可以推出它的子矩阵公式

假设子矩阵从点(x1,y1)到(x2,y2)构成的矩阵,如图3所示;

则可以看成"大矩阵-两小矩阵+多减的小矩阵"，注意:边缘部分

推得: 子矩阵所有元素的和=b[x2][y2]-b[x2][y1-1]-b[x1-1][y2]+b[x1-1][y1-1]

即sum=b[i][j]-b[i][j-x]-b[i-y][j]+b[i-x][j-y]

 图3: