前缀和 1041: 最大子矩阵
给你一个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的正整数。
输出
对于每组数据,输出一个整数,表示子矩阵的最大和。
样例输入
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
样例输出
2474
蓝色区域和等于黑色区域和(大的)减去绿色和红色加上黄色
matrix[i][j]+matrix[i-x][j-y]-matrix[i-x][j]-matrix[i][j-y];
#include<stdio.h>
int T,m,n,x,y;
int matrix[1005][1005];
int main(){
int i,j;
scanf("%d",&T);
while(T--){
scanf("%d %d %d %d",&m,&n,&x,&y);
for(i=0;i<m;i++)
for(j=0;j<n;j++)
scanf("%d",&matrix[i][j]);
for(i=0;i<m;i++)
for(j=0;j<n;j++)
{
if(i-1>=0) matrix[i][j]+=matrix[i-1][j];
if(j-1>=0) matrix[i][j]+=matrix[i][j-1];
if(i-1>=0&&j-1>=0) matrix[i][j]-=matrix[i-1][j-1];
}
int ANS=0,tmp;
for(i=x-1;i<m;i++)
for(j=y-1;j<n;j++)
{if(i-x>=0&&j-y>=0)
tmp=matrix[i][j]+matrix[i-x][j-y]-matrix[i-x][j]-matrix[i][j-y];
else if(i-x>=0) tmp=matrix[i][j]-matrix[i-x][j];
else if(j-y>=0) tmp=matrix[i][j]-matrix[i][j-y];
else tmp=matrix[i][j];
ANS=tmp>ANS?tmp:ANS;
}
printf("%d\n",ANS);
}
return 0;
}