P2216 [HAOI2007]理想的正方形 ( 二维ST表 )

题目链接：点击进入

题目

思路

maxx [ i ] [ j ] ：左上角坐标 ( i , j ) ，边长为 $2^k$ 的正方形的最大值
minn [ i ] [ j ] ：左上角坐标 ( i , j )，边长为 $2^k$ 的正方形的最小值
我们可以枚举 k ，已知长度为 $2^k$ 的正方形可以由长度为 $2^{k-1}$ 的正方形推得，因此我们从 k = 0 开始递推，$2^k$ 的正方形可以推出 $2^{k+1}$ 的正方形，枚举到 $log2(n)-1$ 即可，转移方程：

$maxx[i][j]=getmax(maxx[i][j],maxx[i+(1<<k)][j],maxx[i][j+(1<<k)],maxx[i+(1<<k)][j+(1<<k)])；$

$minn[i][j]=getmin(minn[i][j],minn[i+(1<<k)][j],minn[i][j+(1<<k)],minn[i+(1<<k)][j+(1<<k)]);$

( getmax , getmin 是自定义取最值函数 ，这个转移方程得到的是左上角 ( i , j ) ，长度为 $2^{k+1}$ 的正方形的最大值，如果不明白可以画图看一下，可能会更清楚一点 )

查询函数 ( 取左上角 ( i , j ) ，长度为 n 的正方形的最值 )：

$getmax(maxx[x][y],maxx[x+n-(1<<k)][y],maxx[x][y+n-(1<<k)],maxx[x+n-(1<<k)][y+n-(1<<k)]);$

$getmin(minn[x][y],minn[x+n-(1<<k)][y],minn[x][y+n-(1<<k)],minn[x+n-(1<<k)][y+n-(1<<k)]);$

$k=log2(n)$ ,这里为什么是 x + n - ( 1 << k ) 而不是 x + ( 1<<k ) ? 因为我们之前枚举的时候已经通过 $log2(n)-1$ 得到 $log2(n)$ 了，也就是说对于 maxx [ i ][ j ] 我们已经得到左上角 ( i , j ) 长度 $2^{log2(n)}$ 的正方形里的最值了，但是由于 $log2(n)$ 不一定是整数 ( 我们取 $k=log2(n)$ 时下取整了 ) ，那我们得到的 $2^{log2(n)}$ 长度就可能比 n 小，那么我们就需要再将剩下的那部分考虑上，那么我们可以借鉴一维ST表的思路，取 左边界 + $2^k$ ，与 右边界 - $2^k$ 这两区间的最值，转换到二维，就是取左上角为 $(i,j),(i+n-2^k,j),(i,j+n-2^k),(i+n-2^k,j+n-2^k)$ 长度为 $2^k$ 的四个正方形的最值 )

代码

// Problem: P2216 [HAOI2007]理想的正方形
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P2216
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=1e9+7;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int maxx[1010][1010];
int minn[1010][1010];
int a,b,n,x,ans=inf;
int getmax(int a,int b,int c,int d)
{
	return max(a,max(b,max(c,d)));
}
int getmin(int a,int b,int c,int d)
{
	return min(a,min(b,min(c,d)));
}
int query(int x,int y,int k)
{
	int max1=getmax(maxx[x][y],maxx[x+n-(1<<k)][y],maxx[x][y+n-(1<<k)],maxx[x+n-(1<<k)][y+n-(1<<k)]);
	int min1=getmin(minn[x][y],minn[x+n-(1<<k)][y],minn[x][y+n-(1<<k)],minn[x+n-(1<<k)][y+n-(1<<k)]);
	return max1-min1;
}

int main() 
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	cin>>a>>b>>n;
	for(int i=1;i<=a;i++)
	{
		for(int j=1;j<=b;j++)
		{
			cin>>x;
			maxx[i][j]=minn[i][j]=x;
		}
	}
	// cout<<(1<<1)<<endl;
	int lk=(int)log2(n);
	for(int k=0;k<lk;k++)
		for(int i=1;i+(1<<k)<=a;i++)
			for(int j=1;j+(1<<k)<=b;j++)
				maxx[i][j]=getmax(maxx[i][j],maxx[i+(1<<k)][j],maxx[i][j+(1<<k)],maxx[i+(1<<k)][j+(1<<k)]),
				minn[i][j]=getmin(minn[i][j],minn[i+(1<<k)][j],minn[i][j+(1<<k)],minn[i+(1<<k)][j+(1<<k)]);
	for(int i=1;i+n-1<=a;i++)
		for(int j=1;j+n-1<=b;j++)
			ans=min(ans,query(i,j,lk));
	cout<<ans<<endl;
    return 0;
}