Leetcode 209.长度最小的子数组 By Python

给定一个含有 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