HDU 1559 最大子矩阵

HDU 1559 最大子矩阵

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

思路
这道题和luoguP1719 最大加权矩形很像，不同的是那道题没有确定子矩阵的长和宽，而此题限制了所求矩阵的长和宽。但主要想法都是一样的，将二维数组压缩成一维数组，再用求最大子段和的方法去求最大值。
子矩阵的宽可以在循环时控制，只有子矩阵的宽度等于题中要求时，才去求最大和。而长可以在求最大子段和时控制，可以先求出前缀和，再计算，即y个数的和s=a[i]-a[i-y]
注意有一些加1减1的地方可以代数试试，容易搞错

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;

int n,m,x,y;
int w[1005][1005];
long long int b[1005];

long long int _max(long long int a[])		//求一维数组长度为y的最大和 
{
	long int tep=-1;
	long int s=-1e8;
	for(int i=1;i<=n;i++)
		a[i]+=a[i-1];						//前缀和 
	for(int i=y;i<=n;i++)
	{
		tep=a[i]-a[i-y];					//求y个数的和 
		s=max(s,tep);						//取最大 
	}
	return s;
}

int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		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",&w[i][j]);
		long long int ans=-1e8,s=0;
		for(int i=1;i+x-1<=m;i++)			//循环条件，因为要求宽为x，所以需满足i+x-1<=m 
		{		
			memset(b,0,sizeof(b));			//将辅助数组重新置0 
			for(int j=i;;j++)
			{
				for(int k=1;k<=n;k++)
				{
					b[k]+=w[j][k];			//压缩二维数组 
				}
				if(j-i+1==x)				//此时满足宽为x 
				{
					s=_max(b);				//求和 
					ans=max(ans,s);			//取最大值 
					break;					//退出循环，起始行增加继续 
				}
			}
		}	
		printf("%lld\n",ans);
	}
	return 0;
}