每日一题 LCP24.数字游戏

LCP 24. 数字游戏

题目描述：

小扣在秋日市集入口处发现了一个数字游戏。主办方共有 N 个计数器，计数器编号为 0 ~ N-1。每个计数器上分别显示了一个数字，小扣按计数器编号升序将所显示的数字记于数组 nums。每个计数器上有两个按钮，分别可以实现将显示数字加一或减一。小扣每一次操作可以选择一个计数器，按下加一或减一按钮。

主办方请小扣回答出一个长度为 N 的数组，第 i 个元素(0 <= i < N)表示将 0~i 号计数器 初始 所示数字操作成满足所有条件 nums[a]+1 == nums[a+1],(0 <= a < i) 的最小操作数。回答正确方可进入秋日市集。

由于答案可能很大，请将每个最小操作数对 1,000,000,007 取余。

示例 1：

输入：nums = [3,4,5,1,6,7]

输出：[0,0,0,5,6,7]

解释： i = 0，[3] 无需操作 i = 1，[3,4] 无需操作； i = 2，[3,4,5] 无需操作； i = 3，将 [3,4,5,1] 操作成 [3,4,5,6], 最少 5 次操作； i = 4，将 [3,4,5,1,6] 操作成 [3,4,5,6,7], 最少 6 次操作； i = 5，将 [3,4,5,1,6,7] 操作成 [3,4,5,6,7,8]，最少 7 次操作； 返回 [0,0,0,5,6,7]。

示例 2：

输入：nums = [1,2,3,4,5]

输出：[0,0,0,0,0]

解释：对于任意计数器编号 i 都无需操作。

示例 3：

输入：nums = [1,1,1,2,3,4]

输出：[0,1,2,3,3,3]

解释： i = 0，无需操作； i = 1，将 [1,1] 操作成 [1,2] 或 [0,1] 最少 1 次操作； i = 2，将 [1,1,1] 操作成 [1,2,3] 或 [0,1,2]，最少 2 次操作； i = 3，将 [1,1,1,2] 操作成 [1,2,3,4] 或 [0,1,2,3]，最少 3 次操作； i = 4，将 [1,1,1,2,3] 操作成 [-1,0,1,2,3]，最少 3 次操作； i = 5，将 [1,1,1,2,3,4] 操作成 [-1,0,1,2,3,4]，最少 3 次操作； 返回 [0,1,2,3,3,3]。

提示：

• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^3

思路：

1)刚开始没看明白，以为是新增一个数组，内容为nums[0]~num[0]+len(nums)；然后新数组减去原数组，求和！！有问题，不是这样的！
2)对不起了，我是纯纯CV党。

回到题目，我们需要对 nums\textit{nums}nums 的每个前缀，按照提示 2 中的公式，计算最小距离和。

做法类似 295. 数据流的中位数，用两个堆来维护前 kkk 个数中，nums[i]−i\textit{nums}[i]-inums[i]−i 较小的一半和较大的一半，从而动态维护提示 2 中的「左半之和」以及「右半之和」。

具体来说，用一个大根堆 left维护较小的一半，其元素和为 leftSum；用一个小根堆 right\维护较大的一半，其元素和为 rightSum。遍历 nums[i]，设 b=nums[i]−i，分类讨论：

如果前缀长度是奇数，此时 left 和 right大小相等，我们先把 b插入 left，然后弹出 left 的堆顶，加到 right中。这一操作可以保证，无论 b 是大是小，此时 right的堆顶就是中位数 x。最小距离和为 rightSum−x−leftSum。

如果前缀长度是偶数，此时 left比 right少一个元素，我们先把 bbb 插入 right，然后弹出 right的堆顶，加到 left中。最小距离和为 rightSum−leftSum。
最后，别忘了对 10^9+7取模。

代码：

class Solution:
    def numsGame(self, nums: List[int]) -> List[int]:
        MOD = 1_000_000_007
        ans = [0] * len(nums)
        left = []   # 维护较小的一半，大根堆（小根堆取负号）
        right = []  # 维护较大的一半，小根堆
        left_sum = right_sum = 0
        for i, b in enumerate(nums):
            b -= i
            if i % 2 == 0:  # 前缀长度是奇数
                left_sum -= max(-left[0] - b, 0) if left else 0
                t = -heappushpop(left, -b)
                right_sum += t
                heappush(right, t)
                ans[i] = (right_sum - right[0] - left_sum) % MOD
            else:  # 前缀长度是偶数
                right_sum += max(b - right[0], 0)
                t = heappushpop(right, b)
                left_sum += t
                heappush(left, -t)
                ans[i] = (right_sum - left_sum) % MOD
        return ans

感谢灵茶山艾府