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

题目：

给定整数数组 nums 和整数 k，请返回数组中第 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

方法一：快速排序

1. 快排：选择一个点x，将数组分为左边大于x，右边小于x
2. 判断第k个最大的元素位于哪个区间内
3. 递归排序第k个最大元素所在区间

class Solution {
    int quick_sort(int q[],int l,int r,int k){
        if(l==r) return q[l];
        int i = l-1;
        int j = r+1;
        int x = q[l];

        while(i<j){
            do i++;while(q[i]>x);
            do j--;while(q[j]<x);
            if(i<j){
                int temp = q[j];
                q[j] = q[i];
                q[i] = temp;
            }
        }
        //此时数组被分为左边大于x，右边小于x，判断k在哪个区间内
        int s1 = j-l+1;     //s1为左半段元素个数
        if(k<=s1) {
            return quick_sort(q,l,j,k);
        }else{
            return quick_sort(q,j+1,r,k-s1);
        }
    }
    public int findKthLargest(int[] nums, int k) {
        return quick_sort(nums,0,nums.length-1,k);
    }
}

方法二：维护一个小根堆（PriorityQueue）

class Solution {
    public int findKthLargest(int[] nums, int k) {
        //建立一个小根堆
        PriorityQueue<Integer> minHeap = new PriorityQueue<>();
        for(int i:nums){minHeap.add(i);}    //元素顺序按照自然顺序排列
        //找第k个最大的元素
        while(minHeap.size()-k>0) minHeap.poll();
        return minHeap.peek();
    }
}

方法三：维护一个大根堆

class Solution {
    public int findKthLargest(int[] nums, int k) {
        //建立一个大根堆
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a,b)->b-a);
        for(int i:nums){maxHeap.add(i);}    //数组元素入队，此时元素从大到小排列
        //找到第k个最大的元素
        while(k>1) {maxHeap.poll();k--;}
        return maxHeap.peek();
    }
}

注：

优先队列从大到小排列简洁写法：
PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a,b)->b-a);

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