二维前缀和

直接看这篇: 二维前缀和详解

预处理矩阵a,pre[x][y]为左上角(1,1)到右下角(x,y)矩阵的和:

for(int i=1;i<=n;++i){
        for(int j=1;j<=n;++j){
            pre[i][j]=pre[i][j-1]+pre[i-1][j]-pre[i-1][j-1]+a[i][j];
        }
    }

求左上角(x1,y1)到右下角(x2,y2)矩阵的和:

long long sum=pre[x2][y2]-pre[x1-1][y2]-pre[x2][y1-1]+pre[x1-1][y1-1];



例题:金牌厨师HiLin与HJGG
题意:在一个大矩阵中,求最小的子矩阵,子矩阵和大于k。
解析:子矩阵的大小和子矩阵的最大和具有单调性,有二分查询,二维前缀和来Check。

#include<bits/stdc++.h>
using namespace std;
int a[2010][2010];
int n,k;
long long pre[2010][2010];
bool Check(int m){
    long long sum=0;
    int x1,y1,x2,y2;
    for(int i=1;i<=n-m+1;++i){
        for(int j=1;j<=n-m+1;++j){
            x1=i,y1=j,x2=i+m-1,y2=j+m-1;
            sum=pre[x2][y2]-pre[x1-1][y2]-pre[x2][y1-1]+pre[x1-1][y1-1];
            if(sum>=k) return true;
        }
    }
    return false;
}
int main(){
    long long sum=0;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i){
        for(int j=1;j<=n;++j){
            scanf("%d",&a[i][j]);
            sum+=a[i][j];
        }
    }
    for(int i=1;i<=n;++i){
        for(int j=1;j<=n;++j){
            pre[i][j]=pre[i][j-1]+pre[i-1][j]-pre[i-1][j-1]+a[i][j];
        }
    }
    if(k>sum){
        cout<<"I'm a Gold Chef!"<<endl;
    }
    else {
        int l=1,r=n,ans=n;
        while(l<=r){
            int m=(l+r)/2;
            if(Check(m)){
                ans=min(ans,m);
                r=m-1;
            }
            else l=m+1;
        }
        cout<<ans<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值