【BZOJ1047】[HAOI2007]理想的正方形 （倍增ST表）

[HAOI2007]理想的正方形

题目描述

有一个\(a*b\)的整数组成的矩阵，现请你从中找出一个\(n*n\)的正方形区域，使得该区域所有数中的最大值和最小值的差最小。

输入输出格式

输入格式：

第一行为3个整数，分别表示\(a,b,n\)的值

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

输出格式：

仅一个整数，为\(a*b\)矩阵中所有\(n*n\)正方形区域中的最大整数和最小整数的差值”的最小值。

输入输出样例

输入样例#1：

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）矩阵中的所有数都不超过\(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\)表，和一维的大同小异。

\(mx[k][i][j]\)表示以\((i,j)\)作为左上角，边长为\(2^k\)的正方形内的最大值。

转移：

$mx[k][i][j]=max(max(mx[k-1][i][j],mx[k-1][i][j+(1<<(k-1))]),max(mx[k-1][i+(1<<(k-1))][j],mx[k-1][i+(1<<(k-1))][j+(1<<(k-1))])); $

最小值也是一样。

最后\(N^2\)枚举左上角端点统计答案。

code：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#define ll long long
#define R register
#define N 1005
using namespace std;
template<typename T>inline void read(T &a){
    char c=getchar();T x=0,f=1;
    while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
    while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}
    a=f*x;
}
int a,b,n,x,maxans,minans,ans=0x7fffffffLL,mx[11][N][N],mi[11][N][N];
int main(){
    read(a);read(b);read(n);
    for(R int i=1;i<=a;i++)
        for(R int j=1;j<=b;j++)
            read(x),mi[0][i][j]=mx[0][i][j]=x;
    for(R int k=1;(1<<k)<=n;k++)
        for(R int i=1;i<=a-(1<<k)+1;i++)
            for(R int j=1;j<=b-(1<<k)+1;j++)
                mi[k][i][j]=min(min(mi[k-1][i][j],mi[k-1][i][j+(1<<(k-1))]),min(mi[k-1][i+(1<<(k-1))][j],mi[k-1][i+(1<<(k-1))][j+(1<<(k-1))])),
                mx[k][i][j]=max(max(mx[k-1][i][j],mx[k-1][i][j+(1<<(k-1))]),max(mx[k-1][i+(1<<(k-1))][j],mx[k-1][i+(1<<(k-1))][j+(1<<(k-1))]));
    R int k=(int)log2(n);
    for(R int i=1;i<=a-n+1;i++){
        for(R int j=1;j<=b-n+1;j++){
            minans=0x7fffffff;maxans=-0x7fffffff;
            maxans=max(max(mx[k][i][j],mx[k][i][j+n-(1<<k)]),max(mx[k][i+n-(1<<k)][j],mx[k][i+n-(1<<k)][j+n-(1<<k)]));
            minans=min(min(mi[k][i][j],mi[k][i][j+n-(1<<k)]),min(mi[k][i+n-(1<<k)][j],mi[k][i+n-(1<<k)][j+n-(1<<k)]));
            ans=min(maxans-minans,ans);
        }
    }
    printf("%d\n",ans);	
    return 0;
}