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是一个函数模板,要传对象,我们传的是匿名对象,而下面的是类模板,要传类型,所以不加括号

相关推荐
Black蜡笔小新12 分钟前
自动化AI算法训练服务器DLTM制造业AI质检工作站助力制造业实现AI智检
人工智能·算法·自动化
嵌入式小能手16 分钟前
飞凌嵌入式ElfBoard-进程间的通信之命名管道
linux·服务器·算法
啦哈拉哈38 分钟前
Leetcode题解记录-hot100(81-100)
算法·leetcode·职场和发展
csdn_aspnet1 小时前
Java 霍尔分区算法(Hoare‘s Partition Algorithm)
java·开发语言·算法
王老师青少年编程1 小时前
信奥赛C++提高组csp-s之搜索进阶(搜索剪枝核心思想 )
c++·dfs·csp·信奥赛·搜索剪枝·搜索优化
一拳一个呆瓜1 小时前
【STL】使用 C++ 标准库标头
c++·stl
诸葛务农1 小时前
道路行驶条件下电动汽车永磁电机的有效使用寿命及永磁体的失效和回收再利用(下)
java·开发语言·算法
snow@li1 小时前
AI:理解 大数据、算法、算力、电力、生成式AI、token 之间的关系
大数据·人工智能·算法
小智老师PMP1 小时前
零基础能不能考PMP?零基础专属学习路径+全套扶持体系
学习·算法·职场和发展·软件工程·求职招聘·敏捷流程
王老师青少年编程1 小时前
信奥赛C++提高组csp-s之搜索进阶(搜索剪枝案例实践2)
c++·信奥赛·csp-s·提高组·搜索剪枝·生日蛋糕·最优性剪枝