STL优先级队列,二叉堆leetcode:215.数组中的第k个最大元素

二叉堆在stl里叫做优先级队列,底层的实现方式是vector。
数据结构来说,是一个完全二叉树,按层数来进行填满。

一个节点索引为i,则其子节点的索引为2i和2i+1;
反之其父节点的索引为i/2;

  1. 删除操作:对于堆来说,所有父节点都大于子节点,若此时删除根节点,即删除最大的节点之后,直接将数组末尾,即堆的最末尾元素提到堆顶,然后让堆顶元素与两个节点比较更换较大的上去,如此迭代到换上去的节点重新回到叶子节点的位置,从而维护一个最大/最小堆;
  2. 插入操作同理,直接插入堆末,依次与父节点进行比较,若比父节点大则交换位置,知道满足堆的性质;

STL:priority queue:
常用操作与队列类似:
pop();top();empty();size();push();

最大堆,最小堆:

//升序队列,小顶堆
2 priority_queue <int,vector<int>,greater<int>> pq;
3 //降序队列,大顶堆
4 priority_queue <int,vector<int>,less<int>> pq;

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

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:
利用堆的特性,维护一个大小为k的小顶堆,遍历数组中的元素,每次遍历与堆顶元素比较,若比堆顶元素大则弹出堆顶元素,将该元素压入堆中,到最后堆里面则存储的是数组中k个最大的数,堆顶为k个数中的最小值,即为所求。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int len = nums.size();
        int sub = len-k;
        priority_queue<int,vector<int>,greater<int>> pq;
        
        for(int i = 0;i<k;i++)
        {
            pq.push(nums[i]);
        }
        
        for(int i = k;i<len;i++)
        {
            if(nums[k]>pq.top())
            {
                pq.pop();
                pq.push(nums[k]);
            }
        }
        return pq.top();
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值