BZOJ 1047 [HAOI2007]理想的正方形 二维ST表+压维

Description

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

Input

　　第一行为3个整数，分别表示a,b,n的值第二行至第a+1行每行为b个非负整数，表示矩阵中相应位置上的数。每
行相邻两数之间用一空格分隔。
100%的数据2<=a,b<=1000,n<=a,n<=b,n<=1000

Output

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

Sample Input

5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2

Sample Output

1

HINT

Source

传送门

不就是求方阵一块内的最值吗！

第一眼就感觉裸题啊……这不是二维线段树模板题目吗？！

后来发觉常数以及空间问题……那就二维ST表吧，效果一样而且更好！

结果根据以前处理二维RMQ的经验……发现十分恶俗的是！

我们计算它的空间……

竟然炸了。。。。

好吧，开始以为这种方法不可做，就百度了一下，，

结果都是什么鬼啊……DP+单调队列？

虽然时间看上去是快了好多的……

对不起大牛们。。本菜鸡不会！

回来看题，发现n是固定的，也就是说每次只要知道一个正方形内的最值就好了。

那不就可以压缩维度了嘛……

想象一下查询只会查询正方形，，

每次要求出一个log(区间长)，这个区间长不是次次相等吗。。

那还开两个log干什么！！

最后还有一个小问题，就是A和B不一定相等。。

也就是说原来的不一定是正方形

其实也一样的；我们预处理的时候可以看到，如果2^k>min(A,B)了，就不会做了。

所以直接取min就好了。

好久没打st表，，，感觉都生疏了……

#include<bits/stdc++.h>
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int 
	NMAX=1005,
	inf=2000000000; 
int A,B,n,k;
int stmax[NMAX][NMAX][11],stmin[NMAX][NMAX][11];
int MAX(int a,int b,int c,int d){
	return max(max(max(a,b),c),d);
}
int MIN(int a,int b,int c,int d){
	return min(min(min(a,b),c),d);
}
void RMQ(){
	int N=min(A,B);
	for (int k=1;(1<<k)<=N;k++)
		for (int i=1;i+(1<<k)-1<=A;i++)
			for (int j=1;j+(1<<k)-1<=B;j++)
				stmax[i][j][k]=MAX(
					stmax[i][j][k-1],
					stmax[i+(1<<(k-1))][j][k-1],
					stmax[i][j+(1<<(k-1))][k-1],
					stmax[i+(1<<(k-1))][j+(1<<(k-1))][k-1]),
				stmin[i][j][k]=MIN(
					stmin[i][j][k-1],
					stmin[i+(1<<(k-1))][j][k-1],
					stmin[i][j+(1<<(k-1))][k-1],
					stmin[i+(1<<(k-1))][j+(1<<(k-1))][k-1]);
				
}
int RMQMAX(int x,int y,int x1,int y1){
	int t1=stmax[x][y][k],
	t2=stmax[x1-(1<<k)+1][y][k],
	t3=stmax[x][y1-(1<<k)+1][k],
	t4=stmax[x1-(1<<k)+1][y1-(1<<k)+1][k];
	return max(max(max(t1,t2),t3),t4);
}
int RMQMIN(int x,int y,int x1,int y1){
	int t1=stmin[x][y][k],
	t2=stmin[x1-(1<<k)+1][y][k],
	t3=stmin[x][y1-(1<<k)+1][k],
	t4=stmin[x1-(1<<k)+1][y1-(1<<k)+1][k];
	return min(min(min(t1,t2),t3),t4);
}
int main(){
	A=read(),B=read(),n=read();
	for (int i=1;i<=A;i++)
		for (int j=1;j<=B;j++)
			stmax[i][j][0]=stmin[i][j][0]=read();
	RMQ();
	k=(int)((double)log(n)/(double)log(2));
	int ans=inf;
	for (int i=1;i+n-1<=A;i++)
		for (int j=1;j+n-1<=B;j++)
			ans=min(ans,RMQMAX(i,j,i+n-1,j+n-1)-
					RMQMIN(i,j,i+n-1,j+n-1));
	printf("%d\n",ans);
	return 0;
}