蓝桥与力扣刷题（215 数组中第k个最大元素）

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

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

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

示例 1:

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

示例 2:

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

提示：

• 1 <= k <= nums.length <= 105
• -104 <= nums[i] <= 104

解题思路＋代码：

代码：

class Solution {
    public int findKthLargest(int[] nums, int k) {
        //堆排序中的大根堆  按照降序排序，再返回第k个最大的元素
        // 使用大根堆（PriorityQueue 需要用 lambda 表达式反转默认的最小堆）
        PriorityQueue<Integer> q  = new PriorityQueue<>((a,b) -> b - a); //创建优先队列q
        
        //增强for循环遍历数组，为优先队列添加元素
        for(int num : nums){
            q.add(num);
        }

        // 取出前 k-1 个最大元素
        for(int i = 0; i < k-1; i++){
            q.poll();
        }

         // 第 k 个最大元素
        return q.poll();
    }
}

 总结：一道优先队列（堆排序）的算法题。题目的要求是找到数组中第 k 个最大的元素，可以使用堆排序中大根堆（降序排序）的思想。堆排序中需要创建一个新的队列（Java中的PriorityQueue）来创建队列，（（a,b）-> b - a）lambda对默认的小根堆进行反转，再使用增强for循环来遍历数组，将数组中的元素(q.add()方法)添加到优先队列q中进行排序，再使用for循环移除前面的k-1个元素，最后返回的q.poll()就是所寻找的第k个最大的元素。