题目:
思路:
既然要找第k个大的元素,那么只需要保留前k个最大的元素即可,只要再加入元素,就往这k个元素里面放,比k个元素的最小值小,就不放,比最小值大,就放进去,最后返回的是k个元素的最小值。
这里用到优先队列priorityQueue
- add(E e) 将指定的元素插入此优先级队列。
- clear() 从此优先级队列中移除所有元素。
- peek() 获取但不移除此队列的头;如果此队列为空,则返回 null。
- poll() 获取并移除此队列的头,如果此队列为空,则返回 null。
- size() 返回此 collection 中的元素数。
- toArray() 返回一个包含此队列所有元素的数组。
还有一些功能似乎重复的方法:
- offer,add区别:
一些队列有大小限制,因此如果想在一个满的队列中加入一个新项,多出的项就会被拒绝。
这时新的 offer 方法就可以起作用了。它不是对调用 add() 方法抛出一个 unchecked 异常,而只是得到由 offer() 返回的 false。 - poll,remove区别:
remove() 和 poll() 方法都是从队列中删除第一个元素。remove() 的行为与 Collection 接口的版本相似,
但是新的 poll() 方法在用空集合调用时不是抛出异常,只是返回 null。因此新的方法更适合容易出现异常条件的情况。 - peek,element区别:
element() 和 peek() 用于在队列的头部查询元素。与 remove() 方法类似,在队列为空时, element() 抛出一个异常,而 peek() 返回 null
class KthLargest {
int k;
PriorityQueue<Integer> queue;
public KthLargest(int k, int[] nums) {
this.k=k;
this.queue = new PriorityQueue<>(k);
for(int i=0;i<nums.length;i++){
add(nums[i]);
}
}
public int add(int val) {
if(queue.size()<k){
queue.add(val);
}else{
if(val<queue.peek()){
}else{
queue.poll();
queue.add(val);
}
}
return queue.peek();
}
}
/**
* Your KthLargest object will be instantiated and called as such:
* KthLargest obj = new KthLargest(k, nums);
* int param_1 = obj.add(val);
*/