Black & White--牛客网，二分，前缀和

题目描述
你有一个长度为 n 的 01 串S，你可以执行最多 m 次操作。
对于每次操作，你可以选择一个位置 i 满足 1≤i≤n，翻转这一位的值，0变成1，1变成0。
定义一个 01 串的价值为其中最长连续0的个数和最长连续1的个数的较大值，求S在经过最多m次操作后的最大价值。
输入描述:

* 第一行一个整数 T ，表示接下来有 T 个样例。
* 首先输入n,m，表示S串的长度n和操作次数m，其中1≤n≤100000，0≤m≤1000；
* 接下来输入一个长度为n的字符串S。

输出描述:

一个整数，表示题面上描述的最大价值。

示例1
输入

2
5 1
00101
2 1
01

输出

4
2

说明

第一个串翻转第三个位置，00001的价值为4；第二个串翻转第一个位置，11的价值为2。

分析

用pre数组记录0或1的个数，pre0[i]表示1--i中0的个数，pre1[i]同理。然后用二分寻找答案，对每个寻找的mid进行验证。如何验证？只要判断一下区间内0或1的个数是否小于等于m即可，因为一旦有0或1的个数小于等于m，那么就可以将该区间内的所有数变成0或1。需要注意的是，二分所得到的mid表示长度，而不是位置。

代码

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<stdlib.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int t,n,m;
char s[N];
int pre0[N],pre1[N];//pre0[i]记录1--i有多少个0 
bool judge(int r)
{
	int l=1;
	while(r<=n)//寻找是否有满足要求的长度为mid的序列  
	{
		//0或1的个数小于m，就可以将整个序列全部变成0或1 
		if(pre0[r]-pre0[l-1]<=m||pre1[r]-pre1[l-1]<=m)
			return 1;
		l++;
		r++;
	}
	return 0;
}
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		memset(pre0,0,sizeof(pre0));
		memset(pre1,0,sizeof(pre1));
		for(int i=1;i<=n;i++)
		{
			cin>>s[i];
			if(s[i]=='0')
			{
				pre0[i]=pre0[i-1]+1;
				pre1[i]=pre1[i-1];
			}
			if(s[i]=='1')
			{
				pre0[i]=pre0[i-1];
				pre1[i]=pre1[i-1]+1;
			}
		}
		int left=1,right=n,ans;
		while(left<=right)
		{
			int mid=(left+right)/2;//mid表示长度，而不是位置 
			if(judge(mid))
			{
				left=mid+1;
				ans=mid;
			}
			else
				right=mid-1;
		}
		cout<<ans<<endl;
	}
	return 0;
}