链接: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;
}