class Solution:
def shortestSubarray(self, A: list[int], k: int) -> int:
n = len(A)
# 创建一个列表
dp = [0] * (n + 1)
for i in range(n):
dp[i + 1] = dp[i] + A[i] # 此处dp代表前缀和
ans = n + 1
Q = deque() # Q中放起点坐标
for j in range(n + 1): # 遍历dp
# 若栈顶dp[Q[-1]]比dp[j]大,那它一定不如dp[j]优,因为j作为起点比Q[-1]更短,子数组和更有可能>=K.
while Q and dp[Q[-1]] >= dp[j]:
Q.pop()
# 此时j作为终点,由Q中dp单调递增,左边的dp更有可能使子数组和>=k,但起点越右子数组越短,所以从左往右找符合条件的最右起点,由于后面符合条件的终点(如y-Q[0])不可能比j-Q[0]更短,所以符合条件的起点用完就pop,不符合条件的起点保留看看后面有没有机会。
while Q and dp[j] - dp[Q[0]] >= k:
ans = min(ans, j - Q.popleft())
Q.append(j)
return ans if ans <= n else -1