LeetCode-Python-1004. 最大连续1的个数 III

给定一个由若干 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记录下当前的区间最大长度。

class Solution(object):
    def longestOnes(self, A, k):
        """
        :type A: List[int]
        :type K: int
        :rtype: int
        """
        l = len(A)
        zero = 0
        lo,hi = 0,0
        res = 0
        for hi in range(l):
            if A[hi] == 0:
                zero += 1
            while zero > k:               
                if A[lo] == 0:
                    zero -= 1
                lo += 1
            # print lo, hi
            res = max(res, hi - lo + 1)
            
        return res