矩阵--单调队列

链接:https://ac.nowcoder.com/acm/problem/17060
来源:牛客网

题目描述
矩阵 M 包含 R 行 C 列,第 i 行第 j 列的值为 Mi,j。
请寻找一个子矩阵,使得这个子矩阵的和最大,且满足以下三个条件:
子矩阵的行数不能超过 X 行。
子矩阵的列数不能超过 Y 列。
子矩阵中 0 的个数不能超过 Z 个。
请输出满足以上条件的最大子矩阵和。

输入描述:
第一行输入五个整数 R,C,X,Y,Z。
接下来 N 行,每行输入 M 个整数,第 i 行第 j 列的整数表示 Mi,j。
1 ≤ R,C ≤ 500.
1 ≤ X ≤ R.
1 ≤ Y ≤ C.
1 ≤ Z ≤ R x C.
-109 ≤  Mi,j  ≤ 109
链接:https://ac.nowcoder.com/acm/problem/17060
来源:牛客网

示例1
输入
复制
5 5 3 3 4
0 0 10 0 0
3 4 0 2 3
-1 3 0 -8 3
0 0 32 -9 3
3 0 45 3 0
输出
复制
82

示例2
输入
复制
2 2 2 2 2
-1 -1
-1 -1
输出
复制
0

思路:
如果是一维的区间最大值,在预处理前缀和之后直接单调队列维护即可
本题是二维的,有长宽限制,因此我们采用转换的方式:
枚举矩形的长的两条边,保证上下边之差在x以内,每一次枚举使用单调队列维护长度为y的滑动区间内的最小值,因为是sum[c2]-sum[c1],c1是滑动区间里维护的值,所以维护最小。对所有ans取max即使最终答案

#include<bits/stdc++.h>
using namespace std;
#define int long long 
#define N 510

int g[N][N],cnt[N][N],s[N][N],q[N],sum[N],sum0[N];
int r,c,x,y,z,ans;

void print(int a[][N]){
    for(int i=1;i<=r;i++){
        for(int j=1;j<=c;j++){
            cout << a[i][j] << " ";
        }
        puts("");
    }
}

signed main(){
    scanf("%lld%lld%lld%lld%lld",&r,&c,&x,&y,&z);
    for(int i=1;i<=r;i++)
        for(int j=1;j<=c;j++){
            scanf("%lld",&g[i][j]);
            s[i][j]=s[i-1][j]+g[i][j];
            cnt[i][j]=cnt[i-1][j];
            if(g[i][j]==0) cnt[i][j]+=1;
        }
    
    for(int i=1;i<=r;i++){
        for(int j=i;j<=r&&j-i+1<=x;j++){
            int hh=0,tt=-1;
            memset(q,0,sizeof q);
            memset(sum,0,sizeof sum);
            memset(sum0,0,sizeof sum0);
            q[++tt]=0;
            for(int k=1;k<=c;k++){
                sum[k]=s[j][k]-s[i-1][k]+sum[k-1];
                sum0[k]=cnt[j][k]-cnt[i-1][k]+sum0[k-1];
                while(hh<=tt&&(k-q[hh]>y||sum0[k]-sum0[q[hh]]>z)) hh++;
                if(hh<=tt) ans=max(ans,sum[k]-sum[q[hh]]);
                while(hh<=tt&&sum[q[tt]]>=sum[k]) tt--;
                q[++tt]=k;
            }
        }
    }
    cout << ans << endl;
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值