寻找数组第K大元素的两种方法

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

方法一

基于 jdk 类库 PriorityQueue 实现

class Solution {
    public int findKthLargest(int[] nums, int k) {
        //优先级队列，默认是自然序排序，也即：小顶堆。本题构造大顶堆
        PriorityQueue<Integer> pq = new PriorityQueue<>(Comparator.reverseOrder());
        for (int i = 0; i < nums.length; i++) {
            //将数组中的元素入队列
            pq.add(nums[i]);
        }

        //用于保存第 k 大元素
        int ele = -1;
        while (!pq.isEmpty() && k-- > 0) {
            //不停地出队列，直至获取到第 k 大元素
            ele = pq.poll();
        }
        return ele;       
    }
}

reheap 方式建堆时间复杂度O(N)，出堆时间复杂度为O(logN)。整体时间复杂度可视为O(N)

方法二

基于快排中的分割算法实现，参考：https://www.cnblogs.com/hapjin/p/5587014.html
时间复杂度为 O(N)