滑动窗口7：Leetcode 1004. 最大连续1的个数 III

Posted 2021-12-13 纵横千里，捭阖四方

题目要求：

给定一个由若干 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。

本题也是练习滑动窗口的好问题，官网的解释有点晦涩，我找到的最好的解释是这样的。

这的难点在于我们可以将K个位置从0翻转成1，这里为了好处理，我们做一个转换：

把「最多可以把 K 个 0 变成 1，求仅包含 1 的最长子数组的长度」转换为 「找出一个最长的子数组，该子数组内最多允许有 K 个 0 」。
经过上面的题意转换，我们可知本题是求最大连续子区间，可以使用滑动窗口方法。滑动窗口的限制条件是：窗口内最多有 K 个 0。

 代码思路：

使用 left 和right 两个指针，分别指向滑动窗口的左右边界。
right主动右移：right 指针每次移动一步。当 A[right] 为0，说明滑动窗口内增加了一个 0；
left被动右移：判断此时窗口内 0 的个数，如果超过了 K，则 left指针被迫右移，直至窗口内的 0 的个数小于等于 K 为止。
滑动窗口长度的最大值就是所求。

注意在有0的时候，left到1之间要有两个0

 

 然后继续找：

此时left再移动一次。

class Solution(object):
    def longestOnes(self, A, K):
        N = len(A)
        res = 0
        left, right = 0, 0
        zeros = 0 
        while right < N:
            if A[right] == 0:
                zeros += 1
            while zeros > K:
                if A[left] == 0:
                    zeros -= 1
                left += 1
            res = max(res, right - left + 1)
            right += 1
        return res