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();
}
}