Description
Given a binary array nums, you should delete one element from it.
Return the size of the longest non-empty subarray containing only 1’s in the resulting array. Return 0 if there is no such subarray.
Example 1:
Input: nums = [1,1,0,1]
Output: 3
Explanation: After deleting the number in position 2, [1,1,1] contains 3 numbers with value of 1's.
Example 2:
Input: nums = [0,1,1,1,0,1,1,0,1]
Output: 5
Explanation: After deleting the number in position 4, [0,1,1,1,1,1,0,1] longest subarray with value of 1's is [1,1,1,1,1].
Example 3:
Input: nums = [1,1,1]
Output: 2
Explanation: You must delete one element.
Constraints:
1 <= nums.length <= 10^5
nums[i] is either 0 or 1.
Solution
Use a sliding window to keep track of subarray that contains at most one 0
, if the number of zeros exceeds one, then shrink the window to meet the requirement.
Time complexity:
o
(
n
)
o(n)
o(n)
Space complexity:
o
(
n
)
o(n)
o(n)
Code
class Solution:
def longestSubarray(self, nums: List[int]) -> int:
left = 0
res = 0
window_cnt = 0
for right in range(len(nums)):
if nums[right] == 0:
window_cnt += 1
while window_cnt >= 2:
res = max(res, right - left - 1)
if nums[left] == 0:
window_cnt -= 1
left += 1
return max(res, right - left)