题目
设计一个找到数据流中排序后的第 k 大元素的类(class)。
方法一:优先队列
大小为 k 的优先队列来存储前 k 大的元素,队头为队列中最小的元素,也就是第 k 大的元素。
使用整数 k 和整数流 nums 初始化对象。
PriorityQueue<Integer> pq;
int k;
public KthLargest(int k, int[] nums) {
this.k=k;
pq=new PriorityQueue<Integer>();
for(int x:nums){
add(x);
}
}
将 val 插入数据流 nums 后,返回当前数据流中第 k 大的元素。
public int add(int val) {
pq.offer(val);
if(pq.size()>k){
pq.poll();
}
return pq.peek();
}
复杂度分析
时间复杂度:初始化时间复杂度为: O(nlogk) ,其中 n 为初始化时 nums 的长度;单次插入时间复杂度为: O(logk)。
空间复杂度: O(k)。需要使用优先队列存储前 k大的元素。