给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组,返回 0。
示例:
输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。
进阶:
如果你已经完成了O(n) 时间复杂度的解法, 请尝试 O(n log n) 时间复杂度的解法。
思路
滑动窗口法,设立两个指针i,j,判断nums[i:j+1]跟s的大小关系来决定i,j的走向,详见注释
代码
class Solution:
def minSubArrayLen(self, s, nums):
"""
:type s: int
:type nums: List[int]
:rtype: int
"""
if sum(nums) < s:
return 0
else:
l = 0
r = 0
sum_lr = 0
length = len(nums)
min_length = length + 1 #假设最大的长度为数组的长度+1
while l < length:
if r < length and sum_lr < s: #r没超范围,而且子数组的和小于s
sum_lr += num[r]
r += 1
else:
sum_lr -= nums[l]
l += 1
if sum_lr >= s: #更新长度
min_length = min(r-l,min_length)
return min_length
原文地址:https://www.cnblogs.com/MartinLwx/p/9858570.html
时间: 2024-10-08 19:12:43