最大子矩阵
Time Limit: 30000/10000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5199 Accepted Submission(s): 2747
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
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
Author
lwg
Source
//2017-09-24
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm> using namespace std; const int N = ; int a[N][N], dp[N][N], n, m, x, y; int main()
{
int T;
scanf("%d", &T);
while(T--){
scanf("%d%d%d%d", &n, &m, &x, &y);
for(int i = ; i < n; i++)
for(int j = ; j < m; j++){
scanf("%d", &a[i][j]);
dp[i][j] = a[i][j];
}
for(int i = ; i < n; i++)
for(int j = ; j < m; j++)
dp[i][j] += dp[i-][j];
for(int j = ; j < m; j++)
for(int i = ; i < n; i++)
dp[i][j] += dp[i][j-];
int ans = ;
for(int i = x-; i < n; i++)
for(int j = y-; j < m; j++){
int a = i-x+;
int b = j-y+;
int tmp = dp[i][j];
if(a->=)tmp -= dp[a-][j];
if(b->=)tmp -= dp[i][b-];
if(a->= && b->=)tmp += dp[a-][b-];
ans = max(ans, tmp);
}
printf("%d\n", ans);
} return ;
}