题目描述:
有一个 a×b的整数组成的矩阵,现请你从中找出一个 n×n 的正方形区域,使得该区域所有数中的最大值和最小值的差最小。
输入格式
第一行为三个整数,分别表示 a,b,n的值;
第二行至第 a+1 行每行为 b 个非负整数,表示矩阵中相应位置上的数。
输出格式
输出仅一个整数,为 a×b 矩阵中所有“n×n 正方形区域中的最大整数和最小整数的差值”的最小值。
数据范围
2≤a,b≤1000,
n≤a,n≤b,n≤100,
矩阵中的所有数都不超过 10^9。
输入样例:
5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2
输出样例:
1
分析:
本题代码看起来复杂,写起来却格外的顺,一遍写完,都不用调试,提交就直接AC了,可能是脑中的模型比较清晰吧。
直接说下思路,先对矩阵逐行求区间长度是n的滑动窗口的最大最小值,分别存进最大最小值数组。然后再对求出的最值数组按列再求一次最值,二者之差就是所有n*n正方形区域内的最大整数与最小整数差了。当然,求滑动区间的最值就是使用单调队列实现的。
如图所示,如果要求3*3正方形中的最值,可以先将各行中长度为3的区间的最值存在这个区间开始的格子里,比如上图蓝色的区域,求完后,第一列的前三个格子就存了这个蓝色正方形各行的最值,再竖着对第一行求下长度为3区间的最值存到第一个单元格里,第一个单元格存储的就是整个正方形区域内的最值了。当然我们是先把初始矩形各行长度为n区间的最值存到一个新的数组,然后再对这个数组竖着求最值存入又一个数组来实现的,以避免值被覆盖。实现细节见代码:
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1005;
int n,m,k,g[N][N],mi[N][N],ma[N][N],q[N],r[N],s[N],t[N];
void getmin(int *a,int *b,int m){//将a中区间长度是k区间最小值存入b
int hh = 0,tt = -1;
for(int i = 1;i <= m;i++){
if(hh <= tt && i - q[hh] >= k) hh++;
while(hh <= tt && a[i] <= a[q[tt]]) tt--;
q[++tt] = i;
if(i >= k) b[i - k + 1] = a[q[hh]];
}
}
void getmax(int *a,int *b,int m){//将a中区间长度是k区间最大值存入b
int hh = 0,tt = -1;
for(int i = 1;i <= m;i++){
if(hh <= tt && i - q[hh] >= k) hh++;
while(hh <= tt && a[i] >= a[q[tt]]) tt--;
q[++tt] = i;
if(i >= k) b[i - k + 1] = a[q[hh]];
}
}
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",&g[i][j]);
for(int i = 1;i <= n;i++){//将各行的最值分别存入mi和ma数组
getmin(g[i],mi[i],m);
getmax(g[i],ma[i],m);
}
int res = 1e9;
for(int i = 1;i <= m - k + 1;i++){
for(int j = 1;j <= n;j++) r[j] = mi[j][i];//取一列的最值数组
getmin(r,s,n);//求最值数组列的最值
for(int j = 1;j <= n;j++) r[j] = ma[j][i];
getmax(r,t,n);
for(int j = 1;j <= n - k + 1;j++) res = min(res,t[j] - s[j]);//更新最值之差
}
printf("%d\n",res);
return 0;
}