ZZULI 1895: 985的0-1串难题

1895: 985的0-1串难题
Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 272  Solved: 81

SubmitStatusWeb BoardDescription
985有一个长度为n的0-1串，已知他最多可以修改k次（每次修改一个字符即0->1 或者 1->0），他想知道连续的全1子串最长是多少。

Input
第一行输入一个整数t，代表有t组测试数据。
每组数据第一行输入两个整数n，k分别代笔上面的信息。
注：1 <= t <= 12，1 <= n <= 100000，0 <= k <= 100000。

Output
一个整数代表可以得到的最大长度。

Sample Input
2
6 3
010100
6 2
010100
Sample Output
5
4
HINT

Source
hpu

SubmitStatusWeb Board

#include <cstdio>
#include <iostream>
using namespace std;
const int MAXX=100000+10;
char sr[MAXX];
int n,k;
int num[MAXX];
int init(){
int i=0;
if(sr[0]=='0')
num[0]=1;
for(i=1;i<n;i++){
if(sr[i]=='0')
num[i]=num[i-1]+1;
else
num[i]=num[i-1];
}
}
bool judge(int mid){
int i=0; int j=0;
for( i=mid-1 ,j=0 ; i<n ; i++,j++){
int temp=0;
if(sr[j]=='0')
temp++;
if(num[i]-num[j]+temp<=k)
return true;
}
return false;
}
int main(){
int T; scanf("%d", &T);
while( T-- ){
scanf("%d %d",&n,&k);
scanf("%s",sr);
init();
int l = 0, r = n;
int mid=0;
int ans=0;
while(l <= r){
mid=( l + r ) >> 1;
if( judge(mid) ){
ans=mid;
l=mid+1;
}
else{
r=mid-1;
}
}
printf("%d\n",ans);
}
return 0;
}