Leetcode.215. 数组中的第K个最大元素

题目来源

目录

题目大意

示例一

示例二

解题思路

实现代码及注释

---

题目大意

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

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

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

示例一

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

示例二

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

解题思路

 建立一个大根堆，将数组元素全放入大根堆中按非升序排序，要求第k个最大元素，则将大根堆中第k个元素之前的全弹出，最终将第k个最大元素放在堆顶，输出堆顶元素即可。

实现代码及注释

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        priority_queue<int,vector<int>,less<int>> que;//建立一个大根堆
        for(int i=0;i<n;i++)
        {
            que.push(nums[i]);//将数组都放入大根堆中进行非升序排序
        }
        int t=0;
        while(que.size()>n-k+1)//将比第k个元素大的元素弹出
        {
            que.pop();
        }
        t=que.top();//此时堆顶元素即为数组中第k个最大元素
        return t;
    }
};