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

經典的二分題目
先打個表記錄下來前i個有多少個0
然後二分查找最大的長度 如果查找在那個片斷0的個數小於k滿足如果大於k肯定不滿足了

#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; 
}