深入理解Java PriorityQueue

2021-03-02 Java學習網

Java中PriorityQueue通過二叉小頂堆實現,可以用一棵完全二叉樹表示。本文從Queue接口函數出發,結合生動的圖解,深入淺出地分析PriorityQueue每個操作的具體過程和時間複雜度,將讓讀者建立對PriorityQueue建立清晰而深入的認識。

總體介紹

前面以Java ArrayDeque為例講解了Stack和Queue,其實還有一種特殊的隊列叫做PriorityQueue,即優先隊列。優先隊列的作用是能保證每次取出的元素都是隊列中權值最小的(Java的優先隊列每次取最小元素,C++的優先隊列每次取最大元素)。這裡牽涉到了大小關係,元素大小的評判可以通過元素本身的自然順序(natural ordering),也可以通過構造時傳入的比較器(Comparator,類似於C++的仿函數)。

Java中PriorityQueue實現了Queue接口,不允許放入null元素;其通過堆實現,具體說是通過完全二叉樹(complete binary tree)實現的小頂堆(任意一個非葉子節點的權值,都不大於其左右子節點的權值),也就意味著可以通過數組來作為PriorityQueue的底層實現。

上圖中我們給每個元素按照層序遍歷的方式進行了編號,如果你足夠細心,會發現父節點和子節點的編號是有聯繫的,更確切的說父子節點的編號之間有如下關係:

leftNo = parentNo*2+1

rightNo = parentNo*2+2

parentNo = (nodeNo-1)/2

通過上述三個公式,可以輕易計算出某個節點的父節點以及子節點的下標。這也就是為什麼可以直接用數組來存儲堆的原因。

PriorityQueue的peek()和element操作是常數時間,add(), offer(), 無參數的remove()以及poll()方法的時間複雜度都是log(N)。

方法剖析add()和offer()

add(E e)和offer(E e)的語義相同,都是向優先隊列中插入元素,只是Queue接口規定二者對插入失敗時的處理不同,前者在插入失敗時拋出異常,後則則會返回false。對於PriorityQueue這兩個方法其實沒什麼差別。

新加入的元素可能會破壞小頂堆的性質,因此需要進行必要的調整。

//offer(E e)
public boolean offer(E e) {
if (e == null)//不允許放入null元素
throw new NullPointerException();
modCount++;
int i = size;
if (i >= queue.length)
grow(i + 1);//自動擴容
size = i + 1;
if (i == 0)//隊列原來為空,這是插入的第一個元素
queue[0] = e;
else
siftUp(i, e);//調整
return true;
}

上述代碼中,擴容函數grow()類似於ArrayList裡的grow()函數,就是再申請一個更大的數組,並將原數組的元素複製過去,這裡不再贅述。需要注意的是siftUp(int k, E x)方法,該方法用於插入元素x並維持堆的特性。

//siftUp()
private void siftUp(int k, E x) {
while (k > 0) {
int parent = (k - 1) >>> 1;//parentNo = (nodeNo-1)/2
Object e = queue[parent];
if (comparator.compare(x, (E) e) >= 0)//調用比較器的比較方法
break;
queue[k] = e;
k = parent;
}
queue[k] = x;
}

新加入的元素x可能會破壞小頂堆的性質,因此需要進行調整。調整的過程為:從k指定的位置開始,將x逐層與當前點的parent進行比較並交換,直到滿足x >= queue[parent]為止。注意這裡的比較可以是元素的自然順序,也可以是依靠比較器的順序。

element()和peek()

element()和peek()的語義完全相同,都是獲取但不刪除隊首元素,也就是隊列中權值最小的那個元素,二者唯一的區別是當方法失敗時前者拋出異常,後者返回null。根據小頂堆的性質,堆頂那個元素就是全局最小的那個;由於堆用數組表示,根據下標關係,0下標處的那個元素既是堆頂元素。所以直接返回數組0下標處的那個元素即可

代碼也就非常簡潔:

//peek()
public E peek() {
if (size == 0)
return null;
return (E) queue[0];//0下標處的那個元素就是最小的那個
}

remove()和poll()

remove()和poll()方法的語義也完全相同,都是獲取並刪除隊首元素,區別是當方法失敗時前者拋出異常,後者返回null。由於刪除操作會改變隊列的結構,為維護小頂堆的性質,需要進行必要的調整。

代碼如下:

public E poll() {
if (size == 0)
return null;
int s = --size;
modCount++;
E result = (E) queue[0];//0下標處的那個元素就是最小的那個
E x = (E) queue[s];
queue[s] = null;
if (s != 0)
siftDown(0, x);//調整
return result;
}

上述代碼首先記錄0下標處的元素,並用最後一個元素替換0下標位置的元素,之後調用siftDown()方法對堆進行調整,最後返回原來0下標處的那個元素(也就是最小的那個元素)。重點是siftDown(int k, E x)方法,該方法的作用是從k指定的位置開始,將x逐層向下與當前點的左右孩子中較小的那個交換,直到x小於或等於左右孩子中的任何一個為止

//siftDown()
private void siftDown(int k, E x) {
int half = size >>> 1;
while (k < half) {
//首先找到左右孩子中較小的那個,記錄到c裡,並用child記錄其下標
int child = (k << 1) + 1;//leftNo = parentNo*2+1
Object c = queue[child];
int right = child + 1;
if (right < size &&
comparator.compare((E) c, (E) queue[right]) > 0)
c = queue[child = right];
if (comparator.compare(x, (E) c) <= 0)
break;
queue[k] = c;//然後用c取代原來的值
k = child;
}
queue[k] = x;
}

remove(Object o)

remove(Object o)方法用於刪除隊列中跟o相等的某一個元素(如果有多個相等,只刪除一個),該方法不是Queue接口內的方法,而是Collection接口的方法。由於刪除操作會改變隊列結構,所以要進行調整;又由於刪除元素的位置可能是任意的,所以調整過程比其它函數稍加繁瑣。具體來說,remove(Object o)可以分為2種情況:1. 刪除的是最後一個元素。直接刪除即可,不需要調整。2. 刪除的不是最後一個元素,從刪除點開始以最後一個元素為參照調用一次siftDown()即可。此處不再贅述。

具體代碼如下:

//remove(Object o)
public boolean remove(Object o) {
//通過遍歷數組的方式找到第一個滿足o.equals(queue[i])元素的下標
int i = indexOf(o);
if (i == -1)
return false;
int s = --size;
if (s == i) //情況1
queue[i] = null;
else {
E moved = (E) queue[s];
queue[s] = null;
siftDown(i, moved);//情況2
.
}
return true;
}

相關焦點

  • C++ 優先隊列priority_queue
    頭文件優先隊列 priority_queue 是隊列 queue 的一個變種,頭文件是#include <queue>,使用優先隊列必須要包含這個頭文件。;從定義可以看出,雖然要結構是三個參數,但是後兩個參數帶了默認值,所以針對於普通的數據類型,一般情況下指提供第1個參數就可以了,比如 priority_queue<int> 實際上等價於 priority_queue<int, vector<int>, less<int>>。
  • Java中常用的七個阻塞隊列第二篇DelayQueue源碼介紹
    本文主要內容:通過源碼學習Delayqueue及理解Dqueue並用代碼簡單演示使用場景。本文出自凱哥Java(kaigejava)的《凱哥Java並發系列》之《Java並發編程之隊列》系列的第三篇:《Java中常用的七個阻塞隊列第二篇DelayQueue源碼介紹》Java中常用的幾個隊列中,阻塞隊列還有四個沒介紹。
  • Python 進階:queue 隊列源碼分析
    _qsize() def _qsize(self): return len(self.queue)這個代碼片段挺好理解的,無需分析。 Entries are typically tuples of the form: (priority number, data).
  • java 基礎 之 集合 Map 與Set 與Queue
    Map和Set是我們常常使用的,我們將具體學習一下這個Map與Set,並深入的了解這兩個直接的關係。並具體學習一下Queue的操作。Map與SetSet與Map之間的關係非常密切:Set集合和Map集合的對應關係如下。
  • Python 源碼分析:queue 隊列模塊
    _qsize()    def _qsize(self):        return len(self.queue)這個代碼片段挺好理解的,無需分析。作為隊列,主要得完成入隊與出隊的操作,首先是入隊:class Queue:    ...
  • 要想實現EventQueue,那麼該Queue應有的三種狀態代碼,都在這了
    示例代碼如下:package com. wangwenjun. concurrent. chapter05;import java. util.LinkedList;import static java. lang.
  • 深入理解 Java 線程池!
    import java.util.concurrent.LinkedBlockingQueue;import java.util.concurrent.ThreadFactory;import java.util.concurrent.ThreadPoolExecutor;import java.util.concurrent.TimeUnit;import
  • 深入理解Java關鍵字null
    二、null本身不是對象,也不是Objcet的實例 null本身雖然能代表一個不確定的對象,但就null本身來說,它不是對象,也不確定類型,也不是java.lang.Objectif(null instanceof java.lang.Object) {      System.out.println("null屬於java.lang.Object類型");}else {      System.out.println("null不屬於java.lang.Object類型");}
  • 深入理解Java-Condition
    Condition內部是一個condition queue條件隊列。注意這個隊列的節點Node,就是跟AQS用的是同一個/** First node of condition queue. *///指向隊列頭的指針private transient Node firstWaiter;/** Last node of condition queue.
  • 深入理解Java:註解(Annotation)基本概念
    另外,儘管一些annotation通過java的反射api方法在運行時被訪問,而java語言解釋器在工作時忽略了這些annotation。正是由於java虛擬機忽略了Annotation,導致了annotation類型在代碼中是「不起作用」的; 只有通過某種配套的工具才會對annotation類型中的信息進行訪問和處理。
  • 深入理解Java虛擬機:類加載機制
    如果N的描述符如前面所假設的形式,需要加載的元素類型就是"java.lang.Integer",接著由虛擬機生成一個代表該數組維度和元素的數組對象。如果上面兩步沒有出現任何異常,那麼C在虛擬機中實際上已經成為一個有限的類或接口了,但在解析完成前還要進行符號引用驗證,確認D是否具有對C的訪問權限。如果發現不具備訪問權限,將拋出java.lang.IllegalAccessError異常。
  • java中的Queue隊列的用法
    大家好,歡迎來到雄雄的小課堂,今天給大家分享的是「java中的Queue隊列的用法」 前言:好多人對Queue不是很熟悉,畢竟平時也不怎麼用,遇到集合要麼List要麼map這些常用的,殊不知,java中還有個Queue,今天,我們就來看看Queue的用法。
  • 深入理解Java的動態編譯
    於是帶著這樣的想法,深入學習Java的動態編譯。編寫本文的時候使用的是JDK11。基本原理下面這個很眼熟的圖來源於《深入理解Java虛擬機》前端編譯與優化的章節,主要描述編譯的過程:上圖看起來只有三步,其實每一步都有大量的步驟,下圖嘗試相對詳細地描述具體的步驟(圖比較大難以分割,直接放原圖):實際上,僅僅對於編譯這個過程來說
  • Java WeakHashMap 源碼解析
    一般來說,對於常駐類應用(比如server),隨著時間的增加,所佔用的內存往往會持續上升,如果程序中全部使用強引用,那麼很容易造成內存洩漏,最終導致Out Of Memory (OOM),所以 Java 中提供了除強引用之外的其他三種引用,它們全部位於java.lang.ref包中,下面一一介紹。
  • 深入分析Java中的關鍵字static
    這篇文章就把java中static關鍵字的使用方法的原理進行一個深入的分析。先給出這篇文章的大致脈絡:首先,描述了static關鍵字去修飾java類、方法、變量、代碼塊的方法然後,從底層分析static關鍵字,接下來,給出static的一些使用場景和案例最後,對static進行一個總結,包括和普通變量的區分。
  • 跟我學java編程—深入理解while嵌套
    在D盤Java目錄下,新建「AngleSample.java」文件。用記事本打開「AngleSample.java」文件,輸入以下代碼:代碼結構分析程序功能主要是演示while嵌套循環語句的使用。編譯「AngleSample.java」文件,在命令行窗口輸入「javac AngleSample.java」並執行命令,編譯通過後,在命令行窗口輸入「java AngleSample」運行Java程序,命令行窗口顯示如下信息:
  • 跟我學java編程—深入理解for循環語句
    在D盤Java目錄下,新建「FactorialSample.java」文件。用記事本打開「FactorialSample.java」文件,輸入以下代碼:代碼結構分析程序功能主要是演示for循環語句的使用。
  • 面試中經常被問到 Java 引用類型原理,帶你深入剖析
    super T> queue;     //當該引用被加入到queue中的時候,該欄位被設置為queue中的下一個元素,以形成鍊表結構    volatile Reference next;    //在GC時,JVM底層會維護一個叫DiscoveredList的鍊表,
  • 跟我學java編程—深入理解for語句的嵌套循環
    示例1:用「*」輸出一個菱形圖案,圖案如下: 在D盤Java目錄下,新建「ForSample1.java」文件。用記事本打開「ForSample1.java」文件,輸入以下代碼:代碼結構分析程序功能主要是演示for嵌套循環的使用方法。