给你一个下标从 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)。