【HDU 1559】最大子矩阵（二维树状数组）



最大子矩阵

Time Limit: 30000/10000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4626    Accepted Submission(s): 2398

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<bits/stdc++.h>
#define manx 1002
using namespace std;
int n,m,x,y;
int c[manx][manx];
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int y,int num)
{
    for (int i=x; i<=m; i+=lowbit(i))
        for (int j=y; j<=n; j+=lowbit(j))
            c[i][j]+=num;
}
int Sum(int x,int y)
{
    int sum=0;
    for (int i=x; i>=1; i-=lowbit(i))
        for (int j=y; j>=1; j-=lowbit(j))
            sum+=c[i][j];
    return sum;
}
int main()
{
    int t,a;
    scanf("%d",&t);
    while(t--){
        memset(c,0,sizeof(c));
        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);
                add(i,j,a);
            }
        int ans=0;
        for (int i=x; i<=m; i++)
            for (int j=y; j<=n; j++){
                int sum=Sum(i,j)-Sum(i-x,j)-Sum(i,j-y)+Sum(i-x,j-y);
                ans=max(ans,sum);
            }
        printf("%d\n",ans);
    }
    return 0;
}