【单调队列】[HAOI2007]理想的正方形(C++)

描述

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

输入

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

输出

仅一个整数,为ab矩阵中所有“nn正方形区域中的最大整数和最小整数的差值”的最小值。

样例

输入

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

输出

1





二维单调队列。
代码:

#include
using namespace std;
int n,m,k,l,r;
int a[1001],dl[1001],maxn[1001][1001],minn[1001][1001],maxm[1001],minm[1001];
int ans=INT_MAX;
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[j]);
		l=1;
		r=1;
		dl[l]=1;
		for(int j=2; j<=m; j++) {
			while(l<=r&&a[dl[r]]<=a[j])
				r--;
			while(l<=r&&dl[l]<j-k+1)
				l++;
			r++;
			dl[r]=j;
			if(j-k+1>0)
				maxn[i][j-k+1]=a[dl[l]];
		}
		l=1;
		r=1;
		dl[l]=1;
		for(int j=2; j<=m; j++) {
			while(l<=r&&a[dl[r]]>=a[j])
				r--;
			while(l<=r&&dl[l]<j-k+1)
				l++;
			r++;
			dl[r]=j;
			if(j-k+1>0)
				minn[i][j-k+1]=a[dl[l]];
		}
	}
	for(int i=1;i<=m-k+1;i++)
	{
		for(int j=1;j<=n;j++)
		a[j]=maxn[j][i];
		l=1;
		r=1;
		dl[l]=1;
		for(int j=2; j<=n; j++) {
			while(l<=r&&a[dl[r]]<=a[j])
				r--;
			while(l<=r&&dl[l]<j-k+1)
				l++;
			r++;
			dl[r]=j;
			if(j-k+1>0)
				maxm[j-k+1]=a[dl[l]];
		}
		l=1;
		r=1;
		dl[l]=1;
		for(int j=1;j<=n;j++)
		a[j]=minn[j][i];
		for(int j=2; j<=n; j++) {
			while(l<=r&&a[dl[r]]>=a[j])
				r--;
			while(l<=r&&dl[l]<j-k+1)
				l++;
			r++;
			dl[r]=j;
			if(j-k+1>0)
				minm[j-k+1]=a[dl[l]];
		}
		for(int j=1;j<=n-k+1;j++)
		ans=min(ans,maxm[j]-minm[j]);
	}
	printf("%d\n",ans);

	return 0;
}

你可能感兴趣的:(Mushroom,OJ)