215.数组中的第K个最大元素-C++

题目来源:力扣

题目描述:

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

代码:

cpp 复制代码
    int findKthLargest(vector<int>& nums, int k) {
        //N
        priority_queue<int> pq(nums.begin(),nums.end());
        //K*logN
        while(--k){
            pq.pop();
        }
        return pq.top();
    }

这道题我们可以建大堆,也可以建小堆,这是大堆的玩法

当数据量较大时,尤其是文件这种,是需要建小堆的,下面我们来建小堆

cpp 复制代码
    int findKthLargest(vector<int>& nums, int k) {
        //N
        priority_queue<int,vector<int>,greater<int>> pq(nums.begin(),nums.begin()+k);
        //(N-K)*logK
        for(int i=k;i<nums.size();i++){
            if(nums[i]>pq.top()){
                pq.pop();
                pq.push(nums[i]);
            }
        }
        return pq.top();
    }

这是小堆的玩法,我们对前k-1个数据建堆,然后从第k个数据开始,如果这个数据比堆顶数据要大,就替换这个数据,这样我们得到的就是由整个数组中前k大的数据构建成的小堆,而堆顶自然是第k大的数据

这道题我们其实还可以直接排序过的,大家遇到实在不会的,可以先找个简单的方法试一试

这里有一个问题

我们发现,sort里传的有括号,而下面的优先级队列里没有括号

原因是上面的sort是一个函数模板,要传对象,我们传的是匿名对象,而下面的是类模板,要传类型,所以不加括号

相关推荐
卷无止境12 分钟前
Eigen 库如何借助 OpenMP 加速计算
c++·后端
_清歌44 分钟前
DSpark 深度解读:DeepSeek-V4 如何用「半自回归」把推理速度提升 85%
算法
统计实现局1 小时前
SVD 的三步走:双对角化、Givens 收敛、排序
算法
躬行见万象1 小时前
《VLA 系列》UniLab 强化训练 | G1 机器人 |复现
算法
统计实现局1 小时前
对称不定分解(Bunch-Kaufman):为什么 Cholesky 不够用
算法
统计实现局1 小时前
dqrsl 拆解:拿着 QR 结果能算出哪 5 种东西
算法
卷无止境1 小时前
OpenMPI、MPICH 与 OpenMP:关系、核心概念与架构全解
c++·后端
统计实现局1 小时前
为什么 Cholesky 求逆比 Gauss-Jordan 快一倍——行列式溢出防护详
算法
To_OC12 小时前
LC 994 腐烂的橘子:人人都说是 BFS 入门题,我却写了三遍才过
javascript·算法·leetcode
金銀銅鐵16 小时前
[Python] 扩展欧几里得算法
python·数学·算法