最大连续1的个数III

题目链接

给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 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的个数,若超过k,则移动左边界,每次记录窗口的大小。

class Solution:
    def longestOnes(self, nums: List[int], k: int) -> int:
        ans = 0
        n = len(nums)
        zero = 0
        left, right = 0, 0
        while right < n:
            if nums[right] == 0:
                zero += 1
            while zero > k: # 缩小窗口的大小
                if nums[left] == 0:
                    zero -= 1
                left += 1
            ans = max(ans, right - left + 1)
            right += 1

        return ans