703. 數據流中的第K大元素 --19/11/10

題目:
在這裏插入圖片描述

思路:
既然要找第k個大的元素,那麼只需要保留前k個最大的元素即可,只要再加入元素,就往這k個元素裏面放,比k個元素的最小值小,就不放,比最小值大,就放進去,最後返回的是k個元素的最小值。
這裏用到優先隊列priorityQueue

  1. add(E e) 將指定的元素插入此優先級隊列。
  2. clear() 從此優先級隊列中移除所有元素。
  3. peek() 獲取但不移除此隊列的頭;如果此隊列爲空,則返回 null。
  4. poll() 獲取並移除此隊列的頭,如果此隊列爲空,則返回 null。
  5. size() 返回此 collection 中的元素數。
  6. 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);
 */

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章