题目描述
你有一个长度为 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;
}