【LeetCode】最大连续1的个数 III（python）

题目描述：

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

提示：

1 <= A.length <= 20000 0 <= K <= A.length A[i] 为 0 或 1

代码：

class Solution:
    def longestOnes(self, A: List[int], K: int) -> int:
        left = 0
        right = 0
        length = 0
        while right < len(A):
            if K == 0 and A[right] ==0:
                length = max(length, right - left)
                while left <= right and A[left] != 0:
                    left +=1
                left += 1
            elif A[right] == 0:
                K -= 1
            right += 1
        length = max(length, right - left)
        return length