leetcode209. 長度最小的子數組

給定一個含有 n 個正整數的數組和一個正整數 s ,找出該數組中滿足其和 ≥ s 的長度最小的連續子數組。如果不存在符合條件的連續子數組,返回 0。

示例:
輸入: s = 7, nums = [2,3,1,2,4,3]
輸出: 2
解釋: 子數組 [4,3] 是該條件下的長度最小的連續子數組。
進階:
如果你已經完成了O(n) 時間複雜度的解法, 請嘗試 O(n log n) 時間複雜度的解法。

雙指針,O(n) 時間複雜度:

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        if s > sum(nums):
            return 0
        left, right, res, sum_lr = 0, 0, len(nums)+1, 0  # 雙指針都從第一位出發
        while right < len(nums):
            while sum_lr<s and right<len(nums):  # sum_lr小則右指針右移
                sum_lr += nums[right]
                right += 1
            while sum_lr>=s and left>=0:  # sum_lr大則左指針右移
                res = min(res, right-left)
                sum_lr -= nums[left]
                left += 1
        return res

O(n log n) 時間複雜度,用二分的方法找滑動窗口大小:

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        left, right, res = 0, len(nums), 0
        def helper(size):
            sum_size = 0
            for i in range(len(nums)):
                sum_size += nums[i]
                if i >= size:
                    sum_size -= nums[i-size]
                if sum_size >= s:
                    return True
            return False
        while left<=right:
            mid = (left+right)//2  # 滑動窗口大小
            if helper(mid):  # 如果這個大小的窗口可以那麼就縮小
                res = mid
                right = mid-1
            else:  # 否則就增大窗口
                left = mid+1
        return res
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章