[HAOI2007]理想的正方形

[HAOI2007]理想的正方形

题目描述

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

输入输出格式

输入格式：

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

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

输出格式：

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

输入输出样例

输入样例#1：

5 4 21 2 5 60 17 16 016 17 2 12 10 2 11 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

题解：

如果只考虑一行进行求解，直接用单调队列就好了，推广到多行是，可以发现可以先把每一行用单调队列把每一个点与前n个点缩成一个点，求出maxn合minn，然后再对列进行一次单调队列求maxn和minn，找最小值。

代码：

#include<bits/stdc++.h>
using namespace std; 

const int max_n = 1010;
const int inf = 1e9+7;

int a[max_n][max_n],maxn[max_n][max_n],minn[max_n][max_n],t1[max_n],t2[max_n];
int val[max_n],pos[max_n];
int n,m,k,ans=inf; 

inline 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;
}

inline void pre()
{
	int l,r;
	for(int i=1; i<=n; ++i)
	{
		l=1,r=0;
		for(int j=1; j<=m; ++j)
		{
			while(l<=r && val[r]>=a[i][j]) r--;
			val[++r]=a[i][j]; pos[r]=j;
			if(j-pos[l]==k) l++;
			if(j>=k) minn[i][j]=val[l];
		}
		l=1,r=0;
		for(int j=1; j<=m; ++j)
		{
			while(l<=r && val[r]<=a[i][j]) r--;
			val[++r]=a[i][j]; pos[r]=j;
			if(j-pos[l]==k) l++;
			if(j>=k) maxn[i][j]=val[l];
		}
	}
}

inline void work()
{ 
	int l,r;
	for(int i=k; i<=m; ++i)
	{
		l=1,r=0;
		for(int j=1; j<=n; ++j)
		{
			while(l<=r && val[r]>=minn[j][i]) r--;
			val[++r]=minn[j][i]; pos[r]=j;
			if(j-pos[l]==k) l++;
			if(j>=k) t1[j]=val[l];
		}
		l=1,r=0;
		for(int j=1; j<=n; ++j)
		{
			while(l<=r && val[r]<=maxn[j][i]) r--;
			val[++r]=maxn[j][i]; pos[r]=j;
			if(j-pos[l]==k) l++;
			if(j>=k) t2[j]=val[l];
		}
		for(int j=k; j<=n; ++j) 
		  ans=min(ans,t2[j]-t1[j]);
	}
}

int main()
{
	n=read(); m=read(); k=read();
	for(int i=1; i<=n; ++i)
	  for(int j=1; j<=m; ++j)
	    a[i][j]=read();
	pre();
	work();
	printf("%d\n",ans);
	return 0;
}