CF 846 D. Monitor 滑动窗口,k*k子矩阵最大值

题意:n*m的矩阵,初始全部为白色,Q次操作,第i次操作发生在时间ti,(xi,yi)变为黑色.
问第一次出现k*k黑色子矩阵的时间? n,m<=500, k<=min(n,m). 0<=q<=n*m.


二分时间 每次O(n^4)判断. TLE.. (后面发现正解是二分,然后令此时黑色为1,白色为0,用二维前缀和判断是否有k*k的子矩形即可)
a[x][y] 为(x,y)变为黑色的时间,设f[x][y]为以(x,y)为左上角,边长为k的子矩形的最大值.
现在求出最小的f[x][y]即可. 
求f[x][y] 先用优先队列,滑动窗口的方法求出rg[x][y] (x,y)->(x,y+k-1)的最大值.

f[x][y]=max(rg[x][y],rg[x+1][y],...rg[x+k-1][y]) 竖着在用一次单调队列即可.

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> ii;
const int N=5e2+5,inf=0x3f3f3f3f;
int n,m,k,q,a[N][N];
int rg[N][N],f[N][N];
int main()
{
	memset(a,inf,sizeof(a));
	cin>>n>>m>>k>>q;
	int x,y,t,ans=inf;
	for(int i=1;i<=q;i++)
	{
		scanf("%d%d%d",&x,&y,&t);
		a[x][y]=t;
	} 
	for(int i=1;i<=n;i++)
	{
		priority_queue<ii> q;
		for(int j=m;j>=1;j--)
		{
			while(!q.empty()&&q.top().second-j+1>k)
				q.pop();	
			q.push(ii(a[i][j],j));
			rg[i][j]=q.top().first; 	
		}
	}
	for(int j=1;j<=m;j++)
	{
		priority_queue<ii> q; 
		for(int i=n;i>=1;i--)
		{
			while(!q.empty()&&q.top().second-i+1>k)
				q.pop();
			q.push(ii(rg[i][j],i));
			f[i][j]=q.top().first;
		}
	}
	for(int i=1;i+k-1<=n;i++)
		for(int j=1;j+k-1<=m;j++)
			ans=min(ans,f[i][j]);
	if(ans>1e9)
		ans=-1;
	printf("%d\n",ans);
	return 0;
}