题目描述:
给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组,返回 0。
示例:
输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。
解决方案:双指针(O(n))
代码如下:
class Solution(object):
def minSubArrayLen(self, s, nums):
"""
:type s: int
:type nums: List[int]
:rtype: 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(nlogn))
代码如下:
class Solution(object):
def minSubArrayLen(self, s, nums):
"""
:type s: int
:type nums: List[int]
:rtype: 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
提交结果: