设计一个找到数据流中第 k 大元素的类(class)。注意是排序后的第 k 大元素,不是第 k 个不同的元素。
请实现 KthLargest 类:
KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。
int add(int val) 将 val 插入数据流 nums 后,返回当前数据流中第 k 大的元素。
用优先队列实现最小堆
当堆中元素大于k时 移除堆顶元素
则堆顶元素即为第K大元素
class KthLargest {
Queue<Integer> minheap;
int len;
public KthLargest(int k, int[] nums) {
len = k;
minheap = new PriorityQueue<Integer>();
for(int num : nums){
add(num);
}
}
public int add(int val) {
minheap.offer(val);
if(minheap.size() > len){
minheap.poll();
}
return minheap.peek();
}
}