LeetCode每日一题10.26和至少为k的最短子数组

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