要求
给定一个由若干 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, nums: List[int], k: int) -> int:
l,res = 0,0
stack = []
for r in range(len(nums)):
if nums[r] == 0:
stack.append(r)
if len(stack) > k:
l = stack.pop(0) + 1
else:
res = max(res,r - l + 1)
return res
重点问题
解题思路:
我们固定左边,让后我们不断的扫描列表,当我们遇到0的时候,我们使用列表先将0的下标存储起来,然后当我们存储的0的个数超过了k值的话,我们就将我们遇到的第一个0的下标直接释放,将我们的左标记直接放在释放的下标的下一个位置。然后我们不断的保留历史最佳的长度,知道扫描完整个列表,得到了可翻转k个0最长的连续1的序列。