【LeetCode】209. 长度最小的子数组

官方链接

给定一个含有 n 个正整数的数组和一个正整数 s ，找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组，返回 0。

示例: 

输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。

进阶:

如果你已经完成了O(n) 时间复杂度的解法, 请尝试 O(n log n) 时间复杂度的解法。

方案：（1）左右指针

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        l = 0
        r = 0

        length = len(nums) - 1
        result = len(nums) + 1

        while r <= length:
            tmp_sum = nums[l: r+1]

            if sum(tmp_sum) >= s:
                result = min(result, r - l + 1)
                l += 1

            else:
                r += 1

        return result if result != len(nums) + 1 else 0

此方案复杂度较高，优化之。

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        l = 0

        tmp_sum = 0
        result = len(nums) + 1

        for r in range(len(nums)):
            tmp_sum += nums[r]

            while tmp_sum>=s:
                result = min(result, r - l + 1)
                tmp_sum -= nums[l]

                l += 1

        return result if result != len(nums) + 1 else 0