系列文章目录
【拒绝算法PUA】0x00-位运算
【拒绝算法PUA】0x01- 区间比较技巧
【拒绝算法PUA】0x02- 区间合并技巧
【拒绝算法PUA】0x03 - LeetCode 排序类型刷题
【拒绝算法PUA】LeetCode每日一题系列刷题汇总-2025年持续刷新中
C++刷题技巧总结:
[温习C/C++]0x04 刷题基础编码技巧
LeetCode 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 * 105
1 <= nums[i] <= 109
1 <= k <= 109
输入保证答案一定存在,也就是说一定存在一个操作序列使数组中所有元素都大于等于 k 。
解题方法1 (小顶堆模拟法)
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
class Solution {
public:
struct cmp {
bool operator()(long long a, long long b) {
return a > b; // 大顶堆
}
};
int minOperations(vector<int>& nums, int k) {
int ans = 0;
priority_queue<long long ,vector<long long>, cmp> pq(nums.begin(), nums.end());
while (pq.size() >= 2 && pq.top() < k) {
long long x = pq.top();
pq.pop();
long long y = pq.top();
pq.pop();
auto num = min(x, y) * 2 + max(x, y);
pq.push(num);
ans++;
}
return ans;
}
};
int main(int argc, char **argv) {
vector<int> vec = {1000000000,999999999,1000000000,999999999,1000000000,999999999};
int k = 1000000000;
Solution obj;
int ret = obj.minOperations(vec, k);
cout << ret << endl;
return 0;
}
输出:
2