hdoj 1559 最大子矩阵 [dp]

最大子矩阵

Time Limit: 30000/10000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4171    Accepted Submission(s): 2178

Problem Description

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

 

Input

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

 

Output

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

 

Sample Input

  
  

   
   1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280
  
  

 

Sample Output

  
  

   
   2474
  
  

 

先是遍历每一行，从左至右每相邻y个数求和；

在遍历每一列，从上到下每相邻x个数求和，同时求最大值。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[1010][1010];
int sum[1010][1010];
int main()
{
	int t,n,m,x,y;
	scanf("%d",&t);
	while(t--)
	{
		memset(sum,0,sizeof(sum));
		scanf("%d%d%d%d",&n,&m,&x,&y);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				scanf("%d",&a[i][j]);
				sum[i][j]+=sum[i][j-1]+a[i][j];
				if(j>y)
					sum[i][j]-=a[i][j-y];
				//printf("%d\n",sum[i][j]);
			}
		}
		int ans=0;
		for(int j=y;j<=m;j++)
		{
			int s=0;
			for(int i=1;i<=n;i++)
			{
				s+=sum[i][j];
				if(i>x)
					s-=sum[i-x][j];
				ans=max(ans,s);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
 }