题目链接
思路:排序后二分答案。
#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);
}