【打卡第248道】【堆排序应用】【leetCode高频】:215. 数组中的第K个最大元素

1、题目描述

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

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

2、算法分析

这题使用优先级队列比较好写点。

知识补充：

优先队列PriorityQueue是Queue接口的实现，可以对其中元素进行排序，可以放基本数据类型的包装类（如：Integer，Long等）或自定义的类。对于基本数据类型的包装器类，优先队列中元素默认排列顺序是升序排列。但对于自己定义的类来说，需要自己定义比较器。

常用方法：

peek()//返回队首元素
poll()//返回队首元素，队首元素出队列
add()//添加元素
size()//返回队列元素个数
isEmpty()//判断队列是否为空，为空返回true,不空返回false

PriorityQueue优先级队列默认是小根堆，当然也可以自定义比较器。默认升序排序。

3、代码实现

/**
    优先级队列是：PriorityQueue是基于堆排序，小顶堆升序排序。
    对每个进队的元素进行默认升序排序
    队列里面始终是留下的2个元素，并且入队的元素是按照升序排列的
    最后剩下2个元素，且队尾的第一个元素是第k个大的元素，只需要queue.peek()即可。
 */
class Solution {
    public int findKthLargest(int[] nums, int k) {
        if(k <= 0 || k > nums.length){
            return 0;
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        for(int num:nums){
            queue.offer(num);
            // 队列里面始终是留下的2个元素，并且入队的元素是按照升序排列的
            if(queue.size() > k){
                queue.poll();
            }
        }
        return queue.peek();
    }

}