LeetCode 209. 长度最小的子数组(Python)

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

示例: 

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

思路:滑动窗口,左端点 l 和右端点 r,如果 l 到 r 位置之间的数和大于s则左端点向右移动一位,反之右端点向右移动一位,并记录窗口中数字和大于s时的最小长度。

class Solution(object):
    def minSubArrayLen(self, s, nums):
        """
        :type s: int
        :type nums: List[int]
        :rtype: int
        """
        # 滑动窗口
        l = 0
        r = -1
        cur_sum = 0
        result = len(nums) + 1
        while l < len(nums):
            if cur_sum < s and (r + 1) < len(nums):
                r = r + 1
                cur_sum = cur_sum + nums[r]
            else:
                cur_sum = cur_sum - nums[l]
                l = l + 1
            
            if cur_sum >= s:
                result = min(result, r - l + 1)
            
        if result == len(nums) + 1:
            return 0 
        
        return result
                
发布了88 篇原创文章 · 获赞 98 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/HNU_Csee_wjw/article/details/103429992