215. Kth Largest Element in an Array

最新推荐文章于 2020-05-04 23:26:02 发布

原创最新推荐文章于 2020-05-04 23:26:02 发布 · 113 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种在未排序数组中查找第K大元素的方法，提供了两种实现方式：直接排序和使用优先队列。直接排序法简洁明了，而优先队列法则更适用于大数据量的情况。文章还附带了C++代码示例。

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

Example 1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5

Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

Note:
You may assume k is always valid, 1 ≤ k ≤ array's length.

方法1：直接排序

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        return nums[nums.size()-k];
    }
};

方法2：利用priority_queue

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int> q(nums.begin(),nums.end());
        for(int i=0;i<k-1;i++){
            q.pop();
        }
        return q.top();
    }
};

tips:priority_queue只可以用.pop() .top，没有.front()