题目
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
思路
堆排序
收获
用优先队列实现最大最小堆
注意下列代码中优先队列的创建声明,需要自定义比较函数
代码
class Solution {
public int findKthLargest(int[] nums, int k) {
PriorityQueue<Integer> h = new PriorityQueue<>((n1,n2)->n1-n2);
for(int i:nums){
h.add(i);
if(h.size()>k){
h.poll();
}
}
return h.poll();
}
}
原文地址:https://www.cnblogs.com/Jun10ng/p/12355026.html
时间: 2024-10-11 05:16:35