最长全1串

最长全1串

牛客网：最长全1串
题目描述
给你一个01字符串，定义答案=该串中最长的连续1的长度，现在你有至多K次机会，每次机会可以将串中的某个0改成1，现在问最大的可能答案

输入描述:
输入第一行两个整数N,K，表示字符串长度和机会次数

第二行输入N个整数，表示该字符串的元素

示例1
输入
10 2
1 0 0 1 0 1 0 1 0 1
输出
5

思路：
双指针，一次遍历。O（n）时间复杂度。
牛客网的 python2 运行更快，所以附上了 python2 的代码，如果想用 python3，去掉前两行的 raw_ 即可。

用 $l$ 代表左指针, $r$ 代表右指针。当 $k$ 非 $0$ 时，我们就让右指针一直右移，遇到 $1$ 就直接右移，遇到 $0$ 时从 $k$ 中选一个来填充该位，$k$ 就变成了 $k-1$。当 $k$ 为 $0$ 时，右指针遇到 1 依旧右移，遇到 0 时，需要停下来，因为此时的 $k$ 不够用了，左指针要动了。左指针一定要遇到首个 0 时再停下来，此时左指针和右指针之间的数字长度就是指针 $j$ 的最长全1子串 。记录此时的结果后，我们接着右移右指针，重复上述过程即可。代码更清晰！
其中最后的代码 $res=max(res,j-i)$ 包含了 $k$ 用不完的情况以及数字的最后一位是1的情况。

n,k = list(map(int,raw_input().split()))
num = list(map(int,raw_input().split()))
i,j =0,0
res = 0
while j<n:
    if num[j]==1:
        j += 1
    elif k > 0:
        k -= 1
        j += 1
    else:
        res = max(res,j-i)
        while i<j and num[i]==1:
            i += 1
        j += 1
        i += 1
res = max(res,j-i)
print(res)