jq取第一个子元素为select_长度最小的子数组

题目描述

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

---

分析与解答

法一：O(n)时间复杂度采用双指针之快慢指针法：

# 双指针：快慢指针
# 快慢指针形成一个窗口，计算该窗口内元素和与s间的大小
# 比s大，left += 1，直到比s小
# 比s小，right += 1，直到比s大
class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        left = 0
        total = 0
        res = len(nums) + 1 # 设置一个不可能取到的极大值
        for right, n in enumerate(nums):
            total += n
            while total >= s:
                res = min(res, right - left + 1)
                total -= nums[left]
                left += 1
        return res if res <= len(nums) else 0

法二：O(nlogn)的解法相比于暴力O(n*n)的解法优化在计算窗口和的时候用前缀和+二分法，前缀和是一种常用手法，即计算第 i 到第 j 项的和只需计算前 j 项的和减去前 i - 1 项的和，因此我们计算累计求和即可计算任意窗口的和。这道题外层需要遍历i。直接取别人的题解吧：

class Solution:
    def minSubArrayLen(self, s: int, nums) -> int:
        if nums is None or len(nums) == 0:
            return 0
        sums, min_len = [0], float('inf')
        # 前缀和：累计求和的结果，可看做排序数组 sums[i] 表示 nums 中前 i 个元素的和
        for num in nums:
            sums.append(sums[-1] + num)
        # 所有元素的和比 s 还小
        if sums[-1] < s:
            return 0
        # 我们需要找到最小的j，使得 sums[j]-sums[i] >= s；即在 sums 数组中找到大于等于 s+sums[i] 的第一个元素
        # 这里 sums[j]-sums[i] 表示 nums 中第 i+1 个元素至第 j 个元素的和
        for i in range(len(sums)):
            j = self.findFirstBiggerNumber(s+sums[i], sums)
            # print(j)
            if j >= 0 and j-i < min_len:
                min_len = j-i
        return 0 if min_len == float('inf') else min_len

    def findFirstBiggerNumber(self, s, nums):
        """
        找大于等于给定值的第一个元素，二分查找
        :param s: 给定的值
        :param nums: 排序数组
        :return:
        """
        low, high = 0, len(nums)-1
        while low <= high:
            mid = low + (high - low) // 2
            if nums[mid] >= s:
                if mid == 0 or nums[mid-1] < s:
                    return mid
                # 前半部分还有满足条件的元素
                else:
                    high = mid - 1
            else:
                low = mid + 1
        return -1


作者：su-shuo
链接：https://leetcode-cn.com/problems/minimum-size-subarray-sum/solution/fu-za-du-onlognde-er-fen-cha-zhao-fa-by-su-shuo/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。