老卫带你学---leetcode刷题(1004. 最大连续1的个数)

1004. 最大连续1的个数

问题：

给定一个由若干 0 和 1 组成的数组 A，我们最多可以将 K 个值从 0 变成 1 。

返回仅包含 1 的最长（连续）子数组的长度。

示例 1：

输入：A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出：6
解释：
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 6。
示例 2：

输入：A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出：10
解释：
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 10。

解决：

思想：

用双指针lo, hi维护全部是1的区间，

用hi线性扫描，

如果碰到0， zero += 1

如果已经有k个zero了，说明区间左端应该向右调整到跨越过一个0的位置。

每次循环用res记录下当前的区间最大长度。

python代码：

def lowngestOnes(arr, k):
    l = len(arr)
    zero = 0
    low, high = 0, 0
    res = 0
    for high in range(l):
        if arr[high] == 0:
            zero += 1
        while zero > k:
            if arr[low] == 0:
                zero -= 1
            low += 1
        res = max(res, high - low + 1)
    return res

t=[1,1,1,0,0,0,1,1,1,1,0]
print(lowngestOnes(t,2))