程式碼隨想錄演算法訓練營第二天 | 陣列 209.長度最小的子陣列

leetcode 209.長度最小的子陣列

題目

209.長度最小的子陣列
給定一個含有 n 個正整數的陣列和一個正整數 target 。

找出該陣列中滿足其總和大於等於 target 的長度最小的 連續
子陣列
[numsl, numsl+1, ..., numsr-1, numsr] ，並返回其長度。如果不存在符合條件的子陣列，返回 0 。

示例 1
輸入：target = 7, nums = [2,3,1,2,4,3]
輸出：2
解釋：子陣列 [4,3] 是該條件下的長度最小的子陣列。

示例 2：
輸入：target = 11, nums = [1,1,1,1,1,1,1,1]
輸出：0

解題思路

滑動視窗法

實現程式碼

from typing import List
class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        # 陣列所有元素之和小於target，則子陣列的和一定小於target
        if sum(nums) < target:
            return 0
        
        # 定義雙指標 
        left = 0
        right = 0
        sums = 0    # 儲存連續子陣列的和
        res = float('inf')  # 最小連續子陣列的長度
        # 滿足條件時，嘗試縮小子串的長度，尋找最小的連續子陣列
        while right < len(nums):
            sums = sums + nums[right]
            while sums >= target:                
                res = min(res, right - left + 1)
                sums = sums - nums[left]
                left += 1
            right += 1
        return res

test = Solution()
# nums = [1,1,1]
# target = 4
nums = [2,3,1,2,4,3]
target = 7
print(test.minSubArrayLen(target, nums))