【BZOJ1047】理想的正方形 堆套堆 优先队列

可以粗略的理解为二维的滑动窗口最大最小值

首先我想到的是暴力穷举A-n+1行，每一列对应一个n*B的矩形区域做一维的滑动窗口，时间复杂度O（nABlog（n））妥妥TLE

经过分析如果我们为每一横行开一个优先队列维护横向滑动窗口，加入j列的新元素之后同时处理出A-n+1行中以第j列结尾的n*n矩阵的最大最小值，以最小值为例，首先将n行的优先队列的队首清成合法元素，再开一个维护纵向滑动窗口的优先队列，将各横向队的队首元素放到新队里，滑窗维护即可。

每个元素最多加入横向堆一次，纵向堆每次加入A个元素共加入B-n+1次，故总复杂度为O(ABlog（n））

#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define maxn 1005
#define oo 999999999
int A,B,n;
int a[1005][1005];
void _read(int &x)
{
	char ch=getchar(); x=0;
	while(ch<'0' || ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9')
	{
		x=x*10+ch-'0'; ch=getchar();
	}
	return ;
}
struct state
{
	int x,y;
};
struct cmp1
{
	bool operator () (state x,state y)
	{
		return a[x.x][x.y]>a[y.x][y.y];
	}
};
struct cmp2
{
	bool operator () (state x,state y)
	{
		return a[x.x][x.y]<a[y.x][y.y];
	}
};
priority_queue<state ,vector<state>, cmp1>pq1[1005],pq01;
priority_queue<state ,vector<state>, cmp2>pq2[1005],pq02;
void Init()
{
	scanf("%d%d%d",&A,&B,&n);
	for(int i=1;i<=A;i++)
	{
		for(int j=1;j<=B;j++)
		{
			//scanf("%d",&a[i][j]);
			_read(a[i][j]);
		}
	}
	return ;
}
int ans=oo;

void work()
{
	ans=oo;
	for(int i=1;i<=A;i++)
	{
		for(int j=1;j<=n;j++)
		{
			pq1[i].push((state){i,j}); pq2[i].push((state){i,j});
		}
	}
	state mina,maxa;
	for(int j=1;j+n-1<=B;j++)
	{
		for(int i=1;i<n;i++)//将横行堆堆首合法化加入纵向堆 
		{
			mina=pq1[i].top();
			while(mina.y<j){pq1[i].pop(); mina=pq1[i].top();}
			maxa=pq2[i].top();
			while(maxa.y<j){pq2[i].pop(); maxa=pq2[i].top();}
			pq01.push(mina); pq02.push(maxa);
		}
		for(int i=1;i+n-1<=A;i++)
		{
			mina=pq1[i+n-1].top();//将最新一行的横行堆堆首合法化加入纵向堆 
			while(mina.y<j){pq1[i+n-1].pop(); mina=pq1[i+n-1].top();}
			maxa=pq2[i+n-1].top();
			while(maxa.y<j){pq2[i+n-1].pop(); maxa=pq2[i+n-1].top();}
			pq01.push(mina); pq02.push(maxa);
			
			mina=pq01.top();
			while(mina.x<i || mina.y<j){pq01.pop(); mina=pq01.top();}//维护纵向堆 
			maxa=pq02.top();
			while(maxa.x<i || mina.y<j){pq02.pop(); maxa=pq02.top();}
			ans=min(ans,a[maxa.x][maxa.y]-a[mina.x][mina.y]);
		}
		if(j+n<=B) //将新元素加入横向堆 
		{
			for(int i=1;i<=A;i++)
			{
				pq1[i].push((state){i,j+n});
				pq2[i].push((state){i,j+n});
			}
		}
		while(!pq01.empty())pq01.pop();
		while(!pq02.empty())pq02.pop();
	}
	printf("%d\n",ans);
	return ;
}
int main()
{
	freopen("in.txt","r",stdin);
	Init();
	work();
	return 0;
}