209. 长度最小的子数组

Q:

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

示例:

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

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

A:

nlogn的做法:

class Solution:
    def minSubArrayLen(self, s: int, nums) -> int:
        l=len(nums)
        if not l:
            return 0
        sums=[0 for i in range(l+1)]
        i,sums[1]=1,nums[0]
        res=float('inf')
        for x in nums:
            sums[i]=sums[i-1]+x
            i+=1
        for i in range(l):
            le,ri=i+1,l
            while le<ri:
                mi=le+(ri-le)//2
                if sums[mi]-sums[i]<s:
                    le=mi+1
                else:
                    ri=mi
            # print(i,le,ri)
            if sums[le]-sums[i]>=s:
                res=min(res,le-i)
        return res if res!=float('inf') else 0

n的做法:

class Solution:
    def minSubArrayLen(self, s: int, nums) -> int:
        l=len(nums)
        if not l:
            return 0
        sums,le,ri,res=0,0,0,float('inf')
        while le<l:
            while ri<l and sums<s:
                sums+=nums[ri]
                ri+=1
            if sums>=s:
                res=min(res,ri-le)
            sums-=nums[le]
            le+=1
        return res if res!=float('inf') else 0
posted @ 2019-08-23 15:12  NeoZy  阅读(99)  评论(0编辑  收藏  举报