LeetCode 209. 长度最小的子数组(前缀和+贪心+双指针)

1.题目

给定一个含有 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 = 4, nums = [1,4,4]
输出：1

示例 3：

输入：target = 11, nums = [1,1,1,1,1,1,1,1]
输出：0

提示：

• 1 <= target <= 10^9
• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^5

进阶：

• 如果你已经实现 O(n) 时间复杂度的解法, 请尝试设计一个 O(n log(n)) 时间复杂度的解法。

 2.思路

我看了力扣官方的题解，一是暴力法（但Python会超时），二是前缀和+二分查找，但是个人感觉自己的方法更容易理解，且本身运行效率高于官方题解，主打的就是官方敬酒我不喝。

1.题目要求连续的子数组，那么必然是在nums中取一个区间来计算其总和，涉及到区间和，前缀和算法显然就非常合适，令arr为前缀和数组，那么arr[i] - arr[k]，就能获取[i + 1, k]区间的总和

2.另外题目说明了，nums[i]是正整数，这样arr必然是递增的，这为我们贪心和双指针提供了前提条件

3.要使子数组长度最小，那么这个最小的情况显然出现在当一个子数组恰好大于等于target的时候，我们设置左右指针，当子数组和小于target时，右指针右移增大子数组和，反之，则左指针右移减小子数组和并比较最短长度

 3.代码

class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        """前缀和数组arr初始化"""
        arr = nums[:]
        arr.insert(0, 0)
        size = len(arr)
        """排除nums中最小值也小于target的特殊情况"""
        mini = arr[1]
        for i in range(1, size):
            """排除nums中有值恰好等于target的特殊情况"""
            if arr[i] == target: return 1
            mini = min(mini, arr[i])
            """求前缀和"""
            arr[i] += arr[i - 1]
        """排除nums总和小于等于target的特殊情况"""
        if arr[-1] < target: return 0
        if arr[-1] == target: return size - 1
        """左指针和右指针"""
        right = left = 0
        ans = 10 ** 5 + 1
        """贪心算法，当子数组的总和逼近于target时，才有可能获取最短子数组"""
        while right < size:
            """子数组和小了，则右指针右移，反之，则左指针右移"""
            if arr[right] - arr[left] < target:
                right += 1
            else:
                ans = min(right - left, ans)
                left += 1
        """当ans没有变化时，则不存在等于target的子数组"""
        if ans == 10 ** 5 + 1: return 0
        return ans