【数据结构-堆】力扣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 。

最小堆

cpp 复制代码
class Solution {
public:
    int minOperations(vector<int>& nums, int k) {
        priority_queue<long long, vector<long long>, greater<long long>> q(nums.begin(), nums.end());
        int cnt = 0;
        while(q.top() < k){
            long long x = q.top();
            q.pop();
            long long y = q.top();
            q.pop();
            long long a = min(x,y) * 2 + max(x, y);
            q.push(a);

            cnt++;
        }
        return cnt;
    }
};

这道题我们要找最小的两个数,我们可以用优先队列,并且定义greater使其为最小堆,也就是升序排序。接下来就是进行模拟即可,直到q中的最小元素大于等于k为止。

相关推荐
卷福同学10 分钟前
【AI编程】AI+高德MCP不到10分钟搞定上海三日游
人工智能·算法·程序员
mit6.82412 分钟前
[Leetcode] 预处理 | 多叉树bfs | 格雷编码 | static_cast | 矩阵对角线
算法
皮卡蛋炒饭.34 分钟前
数据结构—排序
数据结构·算法·排序算法
??tobenewyorker1 小时前
力扣打卡第23天 二叉搜索树中的众数
数据结构·算法·leetcode
贝塔西塔2 小时前
一文读懂动态规划:多种经典问题和思路
算法·leetcode·动态规划
众链网络2 小时前
AI进化论08:机器学习的崛起——数据和算法的“二人转”,AI“闷声发大财”
人工智能·算法·机器学习
2 小时前
Unity开发中常用的洗牌算法
java·算法·unity·游戏引擎·游戏开发
飒飒真编程4 小时前
C++类模板继承部分知识及测试代码
开发语言·c++·算法
GeminiGlory4 小时前
算法练习6-大数乘法(高精度乘法)
算法
熬了夜的程序员4 小时前
【华为机试】HJ61 放苹果
算法·华为·面试·golang