hdu1559 最大子矩阵(二维树状数组 模板)

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

 

二维树状数组模板

参考：https://blog.csdn.net/qq_35885746/article/details/89247993

#include <bits/stdc++.h>
using namespace std;
const int N=1e3+20;
int mp[N][N];
int n,m,x,y;
int lowbit(int k)
{
    return k&(-k);
}
void add(int x,int y,int val)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        for(int j=y;j<=m;j+=lowbit(j))
        {
            mp[i][j]+=val;
        }
    }
}
int sum(int x,int y)
{
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        for(int j=y;j>0;j-=lowbit(j))
        {
            ans+=mp[i][j];
        }
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d%d",&m,&n,&x,&y);
        memset(mp,0,sizeof(mp));
        int a;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&a);
                add(i,j,a);
            }
        }
        int k=1;
        int maxx=0;
        for(int i=1;i+x-1<=n;i++)
        {
            for(int j=1;j+y-1<=m;j++)
            {
                int ans=sum(i+x-1,j+y-1)-sum(i-1,j+y-1)-sum(i+x-1,j-1)+sum(i-1,j-1);///容斥
                maxx=max(maxx,ans);
            }
        }
        cout<<maxx<<'\n';
    }
    return 0;
}