Educational Codeforces Round 28 D. Monitor（二分+二维前缀和）

题目链接

思路：排序后二分答案。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2505;
struct node{
	int x,y,t;
}s[maxn*maxn];
int sum[maxn][maxn],a[maxn][maxn],n,m,k,q;
bool cmp(const node &a,const node &b)
{
	return a.t<b.t;
}
bool check(int k)
{
	for(int i=1;i<=n-k+1;++i)
	for(int j=1;j<=m-k+1;++j)
	if(sum[i+k-1][j+k-1]-sum[i-1][j+k-1]-sum[i+k-1][j-1]+sum[i-1][j-1]>=k*k) return true;
	return false;
}
 int main()
 {
 	scanf("%d %d %d %d",&n,&m,&k,&q);
 	for(int i=1;i<=q;++i)
 	scanf("%d %d %d",&s[i].x,&s[i].y,&s[i].t);
 	sort(s+1,s+1+q,cmp);
 	if(q<k*k){
 		printf("-1\n");return 0;
	 }
 	int l=1,r=q;
 	while(l<=r)
 	{
 		int mid=(l+r)>>1;
 		for(int i=1;i<=n;++i) for(int j=1;j<=m;++j) a[i][j]=0;
 		for(int i=1;i<=mid;++i) a[s[i].x][s[i].y]=1;
 		for(int i=1;i<=n;++i)
 		for(int j=1;j<=m;++j)
 		sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
 		if(check(k)) r=mid-1;
		 else l=mid+1; 
	 }
	 if(l==q+1) printf("-1\n");
	 else printf("%d\n",s[l].t);
 }