https://ac.nowcoder.com/acm/contest/893/F
首先预处理出前缀和,a[i]表示这个区间有多少个1
然后二分答案,最对答案进行O(n)验证
只需判断区间内0或1的个数加上m是否不小于当前二分的答案即可 时间复杂度O(nlongn)
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<map>
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
string s;
int a[100001];
int n,m;
int find(int x)
{
for(int i=0;i<=s.size()-x;i++)
{
int num=a[i+x-1]-a[i]+s[i]-'0';
int maxx=max(num,x-num);// 找这段区间内 1 和 0个数的最大值
if(maxx+m>=x)
return 1;
}
return 0;
}
int main()
{
int t;
cin>>t;
while(t--)
{
cin>>n>>m;
cin>>s;
a[0]=s[0]-'0';
for(int i=1;i<s.size();i++)
{
a[i]=a[i-1]+s[i]-'0';// 前缀中1的个数
}
int l=1,r=s.size();
int ans=0;
while(l<=r)
{
int mid=(l+r)/2;
if(find(mid))
{
ans=mid;
l=mid+1;
}
else
r=mid-1;
}
cout<<ans<<endl;
}
return 0;
}