本人新手一枚,希望大佬们及时指导我的错误,万分感谢。
题目地址
LeetCode 75 - 学习计划 - 力扣(LeetCode)全球极客挚爱的技术成长平台
给定一个二进制数组 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。
【1】滑滑滑
step1:计数器,一个指针指向最左端,一个指针遍历。
step2:关键来了,我们每经过一个0,我们将0计数器加一,如果计数已经大于k了,那么我们开始移动左指针缩小窗口,直到0的数量在我们的范围内,然后继续。
class Solution:
def longestOnes(self, nums: List[int], k: int) -> int:
#step1
count_0 = 0
i = 0
max_len = 0
for j in range(len(nums)):
#step2
if nums[j] == 0:
count_0 += 1
if count_0 > k:
max_len = max(max_len, j-i)
while nums[i] != 0:
i += 1
i += 1
count_0 -= 1
max_len = max(max_len, j-i+1)
return max_len
我发现我这写的是真的拉跨啊,j每移动一步都要进行max操作。
【2】通过领悟大佬们的书写方法,我悟了。
思想:只要滑到最后我们得到最大窗口即可!!
主要改动为step2,如果计数已经超过最大数量,那么我们移动i,如果i位置的数为0,那么k再加一。如此下去,我们滑到最后得到的就是最大窗口。
class Solution:
def longestOnes(self, nums: List[int], k: int) -> int:
i = 0
for j in range(len(nums)):
if nums[j] == 0:
k -= 1
if k < 0:
if nums[i] == 0:
k += 1
i += 1
return j-i+1
东方不亮西方亮,大佬们帮我提意见~