給定一個含有 n 個正整數的數組和一個正整數 s ,找出該數組中滿足其和 ≥ s 的長度最小的連續子數組。如果不存在符合條件的連續子數組,返回 0。
示例:
輸入: s = 7, nums = [2,3,1,2,4,3]
輸出: 2
解釋: 子數組 [4,3] 是該條件下的長度最小的連續子數組。
進階:
如果你已經完成了O(n) 時間複雜度的解法, 請嘗試 O(n log n) 時間複雜度的解法。
雙指針,O(n) 時間複雜度:
class Solution:
def minSubArrayLen(self, s: int, nums: List[int]) -> int:
if s > sum(nums):
return 0
left, right, res, sum_lr = 0, 0, len(nums)+1, 0 # 雙指針都從第一位出發
while right < len(nums):
while sum_lr<s and right<len(nums): # sum_lr小則右指針右移
sum_lr += nums[right]
right += 1
while sum_lr>=s and left>=0: # sum_lr大則左指針右移
res = min(res, right-left)
sum_lr -= nums[left]
left += 1
return res
O(n log n) 時間複雜度,用二分的方法找滑動窗口大小:
class Solution:
def minSubArrayLen(self, s: int, nums: List[int]) -> int:
left, right, res = 0, len(nums), 0
def helper(size):
sum_size = 0
for i in range(len(nums)):
sum_size += nums[i]
if i >= size:
sum_size -= nums[i-size]
if sum_size >= s:
return True
return False
while left<=right:
mid = (left+right)//2 # 滑動窗口大小
if helper(mid): # 如果這個大小的窗口可以那麼就縮小
res = mid
right = mid-1
else: # 否則就增大窗口
left = mid+1
return res