描述
有一个ab的整数组成的矩阵,现请你从中找出一个nn的正方形区域,使得该区域所有数中的最大值和最小值的差最小。
输入
第一行为3个整数,分别表示a,b,n的值第二行至第a+1行每行为b个非负整数,表示矩阵中相应位置上的数。每行相邻两数之间用一空格分隔。
100%的数据2<=a,b<=1000,n<=a,n<=b,n<=1000
输出
仅一个整数,为ab矩阵中所有“nn正方形区域中的最大整数和最小整数的差值”的最小值。
样例
输入
5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2
输出
1
二维单调队列。
代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,k,l,r;
int a[1001],dl[1001],maxn[1001][1001],minn[1001][1001],maxm[1001],minm[1001];
int ans=INT_MAX;
int main() {
scanf("%d%d%d",&n,&m,&k);
for(int i=1; i<=n; i++) {
for(int j=1; j<=m; j++)
scanf("%d",&a[j]);
l=1;
r=1;
dl[l]=1;
for(int j=2; j<=m; j++) {
while(l<=r&&a[dl[r]]<=a[j])
r--;
while(l<=r&&dl[l]<j-k+1)
l++;
r++;
dl[r]=j;
if(j-k+1>0)
maxn[i][j-k+1]=a[dl[l]];
}
l=1;
r=1;
dl[l]=1;
for(int j=2; j<=m; j++) {
while(l<=r&&a[dl[r]]>=a[j])
r--;
while(l<=r&&dl[l]<j-k+1)
l++;
r++;
dl[r]=j;
if(j-k+1>0)
minn[i][j-k+1]=a[dl[l]];
}
}
for(int i=1;i<=m-k+1;i++)
{
for(int j=1;j<=n;j++)
a[j]=maxn[j][i];
l=1;
r=1;
dl[l]=1;
for(int j=2; j<=n; j++) {
while(l<=r&&a[dl[r]]<=a[j])
r--;
while(l<=r&&dl[l]<j-k+1)
l++;
r++;
dl[r]=j;
if(j-k+1>0)
maxm[j-k+1]=a[dl[l]];
}
l=1;
r=1;
dl[l]=1;
for(int j=1;j<=n;j++)
a[j]=minn[j][i];
for(int j=2; j<=n; j++) {
while(l<=r&&a[dl[r]]>=a[j])
r--;
while(l<=r&&dl[l]<j-k+1)
l++;
r++;
dl[r]=j;
if(j-k+1>0)
minm[j-k+1]=a[dl[l]];
}
for(int j=1;j<=n-k+1;j++)
ans=min(ans,maxm[j]-minm[j]);
}
printf("%d\n",ans);
return 0;
}