hduoj1559 最大子矩阵(二维前缀和)

hduoj1559 最大子矩阵
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

思路：二维前缀和

#include <stdio.h>
#include <string.h>
#include <algorithm>

const int maxn = 1005;
typedef long long LL;
int a[maxn][maxn];
LL sum[maxn][maxn];

int main()
{
  	int T; scanf("%d",&T);
  	while(T--){
  		memset(sum,0,sizeof(sum));
  		int m,n,x,y;
  		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]);
  				sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + a[i][j];
  			}
  		}
  		LL ans = 0;
		for(int i = x; i <= m; i++){
			for(int j = y; j <= n;j++){
				ans = std::max(ans,sum[i][j] - sum[i - x][j] - sum[i][j - y] + sum[i - x][j - y]);
			}
		}
		printf("%lld\n",ans);	
  	}
    return 0;
}