问题描述
给定一个由若干 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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/max-consecutive-ones-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析及代码实现
这道题跟平常的滑窗差不多的思路,然后就找左右指针移动的条件,这里我加了一个ji来统计翻转的次数,如果翻转次数大于K,那么需要把左指针向右移知道移出去一个翻转元素,然后右指针一直向右走就行了
class Solution:
def longestOnes(self, A: List[int], K: int) -> int:
n = len(A)
left = 0
right = 0
res = 0
ji = 0
while right < n:
if A[right] == 0:
ji += 1
while ji > K:
if A[left] == 0:
ji -= 1
left += 1
res = max(res, right - left + 1)
right += 1
return res