郑轻 oj 1895: 985的0-1串难题

Description

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

思路，我是没想出来，原来就是思维题，太笨了，就是在01串的前面都加上零，记录0的位子，然后再求两个0之间的最长串

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
char a[100000+11];
int pos[100000+11];
int main()
{
    int n,k;
    int ant;
    int u;
    scanf ("%d",&u);
    while (u--)
    {
        scanf ("%d %d",&n,&k);
        scanf ("%s",a);
        ant = 0;
        pos[ant++] = 0;
        for (int i = 0 ; i < n ; i++)
            if (a[i] == '0')
                pos[ant++] = i+1;
        pos[ant++] = n+1;   //前面加个0后面加个零
      //  printf("%d\n",ant);
        if (k >= ant - 1)        //k比0个个数多，则全变为1
        {
            printf ("%d\n",n);
            continue;
        }
        int ans = 0;
        for (int i = k + 1 ; i < ant ; i++)
            {
                ans = max (ans , pos[i] - pos[i-k-1] - 1); //求两个0中间最长的串
              //  printf("%d %d\n",pos[i],pos[i-k-1]);
            }

        printf ("%d\n",ans);
    }
    return 0;
}