给你一个二进制数组 nums
,你需要从中删掉一个元素。
请你在删掉元素的结果数组中,返回最长的且只包含 1 的非空子数组的长度。
如果不存在这样的子数组,请返回 0 。
提示 1:
输入:nums = [1,1,0,1] 输出:3 解释:删掉位置 2 的数后,[1,1,1] 包含 3 个 1 。
示例 2:
输入:nums = [0,1,1,1,0,1,1,0,1] 输出:5 解释:删掉位置 4 的数字后,[0,1,1,1,1,1,0,1] 的最长全 1 子数组为 [1,1,1,1,1] 。
示例 3:
输入:nums = [1,1,1] 输出:2 解释:你必须要删除一个元素。
示例 4:
输入:nums = [1,1,0,0,1,1,1,0,1] 输出:4
示例 5:
输入:nums = [0,0,0] 输出:0
提示:
1 <= nums.length <= 10^5
nums[i]
要么是0
要么是1
。
思路:
类似 LeetCode-Python-487. 最大连续1的个数 II,数据规模暗示 O(N)算法,
题意明示滑动窗口,因为是需要一个动态的在输入数组上的子数组来求答案。
left, right 夹出来一个窗口,用 zeroCnt 统计窗口内 0 的个数,当 zeroCnt 比 1 大时,则代表 left 需要向右移动。
时间复杂度:O(N)
空间复杂度:O(1)
class Solution(object):
def longestSubarray(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if sum(nums) == len(nums):
return len(nums) - 1
zeroCnt = 0
left = 0
res = 0
for right in range(len(nums)):
if nums[right] == 0:
zeroCnt += 1
while zeroCnt > 1:
if nums[left] == 0:
zeroCnt -= 1
left += 1
res = max(res, right - left + 1 - zeroCnt)
return res