1004. Max Consecutive Ones III
- 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