LG-P2216 [HAOI2007]理想的正方形

P2216 [HAOI2007]理想的正方形
题目描述
有一个a*b的整数组成的矩阵，现请你从中找出一个n*n的正方形区域，使得该区域所有数中的最大值和最小值的差最小。

输入格式：
第一行为3个整数，分别表示a,b,n的值

第二行至第a+1行每行为b个非负整数，表示矩阵中相应位置上的数。每行相邻两数之间用一空格分隔。

输出格式：
仅一个整数，为a*b矩阵中所有“n*n正方形区域中的最大整数和最小整数的差值”的最小值。

输入样例：
5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2

输出样例：
1

问题规模

（1）矩阵中的所有数都不超过1,000,000,000

（2）20%的数据2<=a,b<=100,n<=a,n<=b,n<=10

（3）100%的数据2<=a,b<=1000,n<=a,n<=b,n<=100

题解
刚看完题目，首先想到的是ST表。
假设读入：N 行 M 列的矩阵 a[ i ] [ n ] 和一个 k
对每行都求一次，构造一个 f[ i ] [ j ] 表示 a[ i ] [ j+n-1 ~ j ] 的最大值的数组，再对数组的每一列刷一趟ST表就好了。（剩下的O(N^2)扫描）
时间：$N^2*logN$；空间：$N^2$

好像差不多也可以了。

从上面的做法得到启发：这道题目就可以看作对 2*N 个长度为 N 的序列，求每个序列中每段长度为 k 的区间里的最大值。
显然，可以用单调队列优化。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1005;
int n,m,k,a[maxn][maxn],f_min[maxn][maxn],f_max[maxn][maxn],q_min[maxn],q_max[maxn],til_min,hea_min,til_max,hea_max,ans_min[maxn][maxn],ans_max[maxn][maxn];
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[i][j]);
        hea_min=hea_max=1;til_min=til_max=0;
        for (int j=1;j<=m;j++)
        {
            while (hea_min<=til_min&&q_min[hea_min]<=j-k) hea_min++;
            while (hea_max<=til_max&&q_max[hea_max]<=j-k) hea_max++;
            if (hea_min<=til_min) f_min[i][j]=min(a[i][j],a[i][q_min[hea_min]]);else f_min[i][j]=a[i][j];
            if (hea_max<=til_max) f_max[i][j]=max(a[i][j],a[i][q_max[hea_max]]);else f_max[i][j]=a[i][j];
            while (hea_min<=til_min&&a[i][q_min[til_min]]>=a[i][j]) til_min--;
            while (hea_max<=til_max&&a[i][q_max[til_max]]<=a[i][j]) til_max--;
            q_min[++til_min]=j;
            q_max[++til_max]=j;
        }
    }
    for (int j=k;j<=m;j++)
    {
        hea_min=hea_max=1;til_min=til_max=0;
        for (int i=1;i<=n;i++)
        {
            while (hea_min<=til_min&&q_min[hea_min]<=i-k) hea_min++;
            while (hea_max<=til_max&&q_max[hea_max]<=i-k) hea_max++;
            if (hea_min<=til_min) ans_min[i][j]=min(f_min[i][j],f_min[q_min[hea_min]][j]);else ans_min[i][j]=f_min[i][j];
            if (hea_max<=til_max) ans_max[i][j]=max(f_max[i][j],f_max[q_max[hea_max]][j]);else ans_max[i][j]=f_max[i][j];
            while (hea_min<=til_min&&f_min[q_min[til_min]][j]>=f_min[i][j]) til_min--;
            while (hea_max<=til_max&&f_max[q_max[til_max]][j]<=f_max[i][j]) til_max--;
            q_min[++til_min]=i;
            q_max[++til_max]=i;
        }
    }
    int ans=999999999;
    for (int i=k;i<=n;i++)
     for (int j=k;j<=m;j++) ans=min(ans_max[i][j]-ans_min[i][j],ans);
    printf("%d\n",ans);
    return 0;
}