LeetCode 3066. 超过阈值的最少操作数 II

3066. 超过阈值的最少操作数 II

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。

一次操作中，你将执行：

• 选择 nums 中最小的两个整数 x 和 y 。
• 将 x 和 y 从 nums 中删除。
• 将 min(x, y) * 2 + max(x, y) 添加到数组中的任意位置。

注意，只有当 nums 至少包含两个元素时，你才可以执行以上操作。

你需要使数组中的所有元素都大于或等于 k ，请你返回需要的 最少 操作次数。

示例 1：

输入：nums = [2,11,10,1,3], k = 10
输出：2
解释：第一次操作中，我们删除元素 1 和 2 ，然后添加 1 * 2 + 2 到 nums 中，nums 变为 [4, 11, 10, 3] 。
第二次操作中，我们删除元素 3 和 4 ，然后添加 3 * 2 + 4 到 nums 中，nums 变为 [10, 11, 10] 。
此时，数组中的所有元素都大于等于 10 ，所以我们停止操作。
使数组中所有元素都大于等于 10 需要的最少操作次数为 2 。

示例 2：

输入：nums = [1,1,2,4,9], k = 20
输出：4
解释：第一次操作后，nums 变为 [2, 4, 9, 3] 。
第二次操作后，nums 变为 [7, 4, 9] 。
第三次操作后，nums 变为 [15, 9] 。
第四次操作后，nums 变为 [33] 。
此时，数组中的所有元素都大于等于 20 ，所以我们停止操作。
使数组中所有元素都大于等于 20 需要的最少操作次数为 4 。

提示：

• 2 <= nums.length <= 2 * 10^5
• 1 <= nums[i] <= 10^9
• 1 <= k <= 10^9
• 输入保证答案一定存在，也就是说一定存在一个操作序列使数组中所有元素都大于等于 k 。

---

提示 1

Use priority queue to keep track of minimum elements.

---

提示 2

Remove the minimum two elements, perform the operation, and insert the resulting number into the priority queue.

解法1：最小堆（优先队列）

Java版：

class Solution {
    public int minOperations(int[] nums, int k) {
        PriorityQueue<Long> q = new PriorityQueue<>((a, b) -> {
            return (int) (a - b);
        });
        for (int num : nums) {
            q.offer((long) num);
        }
        int cnt = 0;
        while (q.size() >= 2) {
            long x = q.poll();
            if (x >= k) {
                break;
            }
            long y = q.poll();
            q.offer(x * 2 + y);
            cnt++;
        }
        return cnt;
    }
}

Python3版：

class Solution:
    def minOperations(self, nums: List[int], k: int) -> int:
        heapify(nums)
        cnt = 0
        while len(nums) >= 2:
            x = heappop(nums)
            if x >= k:
                break
            y = heappop(nums)
            heappush(nums, x * 2 + y)
            cnt += 1
        return cnt

复杂度分析

• 时间复杂度：O(nlogn)，其中 n 为 nums 的长度。
• 空间复杂度：O(n) 或 O(1)。