AcWing 1091 理想的正方形

题目描述：

有一个 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;
}