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

相关推荐
Tairitsu_H18 分钟前
[LC优选算法#18] 前缀和 | 除⾃⾝以外数组的乘积 | 和为K的⼦数组 | 和可被K整除的⼦数组
算法·前缀和·哈希算法
鹏易灵35 分钟前
C++——8.移动语义初探(移动构造、移动赋值)
开发语言·c++·php
凯瑟琳.奥古斯特1 小时前
力扣1008:前序重建BST
开发语言·c++·算法·leetcode·职场和发展
Alvin's Tech Blog1 小时前
内存泄漏会导致操作系统永久丢失内存吗?它会导致应用程序崩溃吗?
c++·内存管理·内存泄漏
会周易的程序员1 小时前
使用 LuaBridge 封装 C++ 日志库 microLog 为 Lua 模块
c++·物联网·junit·lua·日志·iot·aiot
wbs_scy2 小时前
仿 muduo 高并发服务器项目:从 timerfd 到时间轮实现定时任务机制
linux·服务器·c++
aqiu1111112 小时前
【算法日记 13】LeetCode 49. 字母异位词分组:哈希表的进阶“降维打击”
算法·leetcode·散列表
WL学习笔记2 小时前
链式队列(数据结构)
前端·数据结构·jquery
KaMeidebaby2 小时前
卡梅德生物技术快报|纳米抗体技术全套实操流程:AFB1 全合成文库淘选 + 分子对接定点突变参数手册
人工智能·python·tcp/ip·算法·机器学习
巴糖2 小时前
从做早餐理解 DAG 与拓扑排序:任务依赖图的核心逻辑
算法