双指针 - 单调队列 - 二分
这题有很多做法,单调队列/双指针+二分
这里讲一个最简单的,既不用双指针也不用单调队列
首先,输入时我们把可以住的房间都给筛出来,并储存下它们的位置
然后,由于题目给定了需要 个房间,我们就枚举长度为 的连续区间,用二分查找找出最靠近中间的的那间房(让农夫住在这里),并算出离两端的距离。
最后取最小值即可
#include<cstdio>
#include<iostream>
using namespace std;
const int Maxn=100000+10,inf=0x3f3f3f3f;
int n,k,cnt,ans=inf;
int a[Maxn];
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d%d",&n,&k); // 这里 k 没有加1,是为了后面计算的方便
char s[Maxn];
scanf("%s",s+1);
for(int i=1;i<=n;++i)
if(s[i]=='0')a[++cnt]=i; // 筛出空房
n=cnt;
for(int i=1;i+k<=n;++i)
{
int l=i,r=i+k;
int tmp=(a[l]+a[r])>>1;
while(l<r)
{
int mid=(l+r)>>1;
++mid;
if(a[mid]<=tmp)l=mid;
else r=mid-1;
}
for(int j=min(l-2,i);j<=min(l+2,i+k);++j) // 为保险起见,我们也要查看一下相邻的几个房间
ans=min(ans,max(a[j]-a[i],a[i+k]-a[j]));
}
printf("%d\n",ans);
return 0;
}