862. 和至少为 K 的最短子数组
原始题目链接:https://leetcode.cn/problems/shortest-subarray-with-sum-at-least-k/
给你一个整数数组 nums 和一个整数 k ,找出 nums 中和至少为 k 的 最短非空子数组 ,并返回该子数组的长度。如果不存在这样的 子数组 ,返回 -1 。
子数组 是数组中 连续 的一部分。
示例 1:
输入:nums = [1], k = 1
输出:1
示例 2:
输入:nums = [1,2], k = 4
输出:-1
示例 3:
输入:nums = [2,-1,2], k = 3
输出:3
解题思路:
题目要求需要找到两个端点,两个端点的差值至少为k,并且距离最短。用一个单调队列来存储前缀和的下标,这里队列是单调递增序列。
代码实现:
class Solution:
def shortestSubarray(self, nums: List[int], k: int) -> int:
# 单调递增的队列,存储的是前缀和队列的下标
queue = collections.deque()
ans = float('inf')
# 计算前缀和
pre_sum = [0]
for num in nums:
pre_sum.append(pre_sum[-1] + num)
# 遍历前缀和
for i, cur_i_sum in enumerate(pre_sum):
# 在把前缀和的下标加入单调队列前
# 查找当前位置作为尾节点,前缀和中元素作为头节点
# 相差至少为k,更新最小值
while queue and cur_i_sum - pre_sum[queue[0]] >= k:
ans = min(ans, i - queue[0])
# 因为当前元素作为尾节点和前缀和最开头的节点的差值为k了
# 所以当前元素的后面没遍历的节点也可以和前缀和最开头的节点相差为k
# 题目要求最短,并且队列是单调递增的,所以从队头元素开始删除
queue.popleft()
# 在把前缀和的下标加入单调队列前
# 查找在当前元素之前的前缀和比以当前元素作为前缀和大的位置
# 因为后面的数和当前元素相差至少k时,之前的元素也满足
# 但长度大于当前的位置,题目要求求最短,所以删除前缀和比以当前元素作为前缀和大的位置
# 因为是队列是单调递增的,所以从队尾删除
while queue and pre_sum[queue[-1]] >= cur_i_sum:
queue.pop()
queue.append(i)
return ans if ans != float('inf') else -1
参考文献:
https://leetcode.cn/problems/shortest-subarray-with-sum-at-least-k/solution/python-dan-diao-dui-lie-bi-guan-fang-ti-cywko/