[先写一种方法] 用堆实现:
小顶堆,小的元素会浮上去,大的沉下去,最后里面存放k个最大的,那堆顶就是我们要找的第K大的元素。
知识点:
优先队列定义:PriorityQueue pq = new PriorityQueue<>()
插入元素:pq.offer() 失败返回null
删除队首元素:pq.poll() 失败返回null
返回堆顶元素:pq.peek() 失败返回null
class Solution {
public int findKthLargest(int[] nums, int k) {
// 定义二叉堆<优先队列> 二叉小顶堆实现
// 优先队列的作用是能保证每次取出的元素都是队列中权值最小的
PriorityQueue<Integer> pq = new PriorityQueue<>();
for(int e:nums){
// offer(E e)是向优先队列中插入元素,插入失败时抛出异常,后则则会返回false
pq.offer(e);
// 堆的大小:size()
if(pq.size()>k){
// 保证最后留下k个元素,最后的堆顶是整个堆中最小的值,也就是第K个元素
// poll()方法是获取并删除队首元素,当方法失败时返回null。
pq.poll();
}
}
// peek()获取但不删除队首元素,也就是队列中权值最小的那个元素
return pq.peek();
}
}