LeetCode 1004. Max Consecutive Ones III 解题报告（python）

1004. Max Consecutive Ones III

1004. Max Consecutive Ones III python solution

题目描述

Given an array A of 0s and 1s, we may change up to K values from 0 to 1.
Return the length of the longest (contiguous) subarray that contains only 1s.

解析

连续的子数组里面，最多可以变换K个0，可以等价于一个[i,j]的窗口里面的0的个数小于K个。

如果窗口[i,j]里面的0的个数小于K个，那么这个窗口一定是合法。
如果窗口[i,j]里面的0的个数已经大于K个了，那么这个窗口就不合法，我们需要把i向前推进，直到窗口里面的0的个数小于K个。

所以i只有在窗口内0的个数不满足要求时才会推进，j是一直推进的。在这个过程中，窗口的长度是不会减小的。会先增大到最佳长度，然后不变，直到最后结束。

class Solution:
    def longestOnes(self, A, K):
        res = i = 0
        for j in range(len(A)):
            K -= A[j] == 0
            if K < 0:
                K += A[i] == 0
                i += 1
            res = j - i + 1
        return res
        

Reference

https://leetcode.com/problems/max-consecutive-ones-iii/discuss/247564/JavaC%2B%2BPython-Sliding-Window