題目:
思路:
既然要找第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);
*/