1004. 最大连续1的个数 III
每日几道leetcode刷刷题!
lc1004
题目描述
给定一个二进制数组 nums 和一个整数 k,如果可以翻转最多 k 个 0 ,则返回 数组中连续 1 的最大个数 。
示例 1:
输入:nums = [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:
输入:nums = [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。
代码
class Solution:
def longestOnes(self, nums: List[int], k: int) -> int:
start = 0
flag = 0
max_len = 0
for end in range(len(nums)):
if nums[end] == 0: #窗口有0时,flag+1,flag记录窗口有几个零
flag += 1
#当flag>k时,如果nums[start]==0,左移start后,移动的窗口就减少一个0,直到窗口里面只有k个0
while flag > k:
if nums[start] == 0:
flag -= 1 #flag-1
start += 1 #同时strat左移
max_len = max(max_len, end - start + 1)
return max_len
总结
类似题目 传送门
像这种连续的数字,字符串,可以考虑滑动窗口解答,
- 设置窗口的两端,从左到右遍历;
- 当窗口里面的东西不满足要求时,进行操作(while xx条件:左端点移动),直到满足
- 最后输出结果