题目如下
利用单调队列+前缀和
小兔子有 n 颗排成一列的萝卜炸弹,每颗炸弹的伤害值不同。小兔子可以一次性拿最多连续的 m 颗炸弹(至少拿一颗),目标是选择连续的若干颗炸弹,使得它们的总伤害最大。我们需要找出这个最大的总伤害值。
解题思路
这是一个典型的滑动窗口最大和问题,可以通过前缀和 + 单调队列的方法高效解决。具体步骤如下:
计算前缀和数组:
前缀和数组 prefix 定义为 prefix[i] = nums[0] + nums[1] + ... + nums[i-1]。
这样,子数组 nums[i..j] 的和可以表示为 prefix[j+1] - prefix[i]。
单调队列维护最小值:
我们需要找到 prefix[j] - prefix[i] 的最大值,其中 j - i <= m。
这等价于对于每个 j,找到 i 在 [j-m, j-1] 范围内使得 prefix[i] 最小。
因此,可以用单调队列维护一个长度为 m 的窗口中的 prefix[i] 最小值。
遍历并更新最大值:
遍历前缀和数组,用单调队列维护当前窗口的最小 prefix[i],并计算 prefix[j] - prefix[i] 的最大值。
代码如下
谢谢观看!!!