【前缀和】最大子矩阵

这个会(大概?)
打来看看
顺便做个数据储备

Description

给你一个m×n的整数矩阵,在上面找一个x×y的子矩阵,使子矩阵中所有元素的和最大。

容斥原理的牛逼应用
pre[i][j]表示以(i,j)右下角的矩阵的前缀和

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<iostream>
#include<cstring>
using namespace std;
int main(){
    int pre[1005][1005],t,m,n,x,y,a[1005][1005];
    cin>>t;
    while(t>0){
        t--;
        cin>>m>>n>>x>>y;
        memset(pre,0,sizeof(pre));
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                cin>>a[i][j];
                pre[i][j]=pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]+a[i][j];
            }
        }
        long long maxn=-9999999999,tmp=0;
        for(int i=x;i<=m;i++){
            for(int j=y;j<=n;j++){
                tmp=pre[i][j]-pre[i-x][j]-pre[i][j-y]+pre[i-x][j-y];
                if(tmp>maxn)
                    maxn=tmp;
            }
        }
        cout<<maxn<<endl;
    }
    return 0;
}
posted @ 2021-11-25 19:35  Jedi_Pz  阅读(77)  评论(0)    收藏  举报