给你一个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
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1010;
int a[maxn][maxn];
long long s[maxn][maxn];
int main()
{
int t;
cin>>t;
// memset(s,0,sizeof(s)); ??
while(t--)
{
int m,n,x,y;
cin>>m>>n>>x>>y;
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
cin>>a[i][j];
}
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
a[i][j]=a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1]; //原错 s[i][j]=a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];
//注意a[i][i]依赖于前边前缀和推出,像a[i-1][j]..开始考虑是累加过大,s[i][i]会比a[i][i]大很多,但改后没有了递推联系,因而错误
}
int ans=0;
for(int i=x;i<=m;i++)
for(int j=y;j<=n;j++)
{
ans=max(ans,a[i][j]-a[i-x][j]-a[i][j-y]+a[i-x][j-y]);//原错 ans=max(ans,s[i][j]-s[i-x][j]-s[i][j-y]+s[i-x][j-y]);
}
//也可以如下的写法
/*for(int i = 1; i<=m-x+1; i++) {
for(int j = 1; j<=n-y+1; j++) {
ans = max(ans,s[i+x-1][j+y-1]-s[i+x-1][j-1]-s[i-1][j+y-1]+s[i-1][j-1]);
}
}*/
printf("%d\n",ans);
}
return 0;
}