詳解Java中PriorityQueue的作用和源碼實(shí)現(xiàn)
引言
前面文章我們講解了ArrayBlockingQueue和LinkedBlockingQueue源碼,這篇文章開始講解PriorityQueue源碼。從名字上就能看到ArrayBlockingQueue是基于數(shù)組實(shí)現(xiàn)的,而LinkedBlockingQueue是基于鏈表實(shí)現(xiàn),而PriorityQueue是基于什么數(shù)據(jù)結(jié)構(gòu)實(shí)現(xiàn)的,看不出來,好像是實(shí)現(xiàn)了優(yōu)先級(jí)的隊(duì)列。
由于PriorityQueue跟前幾個(gè)阻塞隊(duì)列不一樣,并沒有實(shí)現(xiàn)BlockingQueue接口,只是實(shí)現(xiàn)了Queue接口,Queue接口中定義了幾組放數(shù)據(jù)和取數(shù)據(jù)的方法,來滿足不同的場(chǎng)景。

| 操作 | 拋出異常 | 返回特定值 |
|---|---|---|
| 放數(shù)據(jù) | add() | offer() |
| 取數(shù)據(jù)(同時(shí)刪除數(shù)據(jù)) | remove() | poll() |
| 查看數(shù)據(jù)(不刪除) | element() | peek() |
這兩組方法的區(qū)別是:
- 當(dāng)隊(duì)列滿的時(shí)候,再次添加數(shù)據(jù),add()會(huì)拋出異常,offer()會(huì)返回false。
- 當(dāng)隊(duì)列為空的時(shí)候,再次取數(shù)據(jù),remove()會(huì)拋出異常,poll()會(huì)返回null。
PriorityQueue也會(huì)有針對(duì)這幾組放數(shù)據(jù)和取數(shù)據(jù)方法的具體實(shí)現(xiàn)。
類結(jié)構(gòu)
先看一下PriorityQueue類里面有哪些屬性:
public class PriorityQueue<E>
extends AbstractQueue<E>
implements java.io.Serializable {
/**
* 數(shù)組初始容量大小
*/
private static final int DEFAULT_INITIAL_CAPACITY = 11;
/**
* 數(shù)組,用于存儲(chǔ)元素
*/
transient Object[] queue; // non-private to simplify nested class access
/**
* 元素個(gè)數(shù)
*/
private int size = 0;
/**
* 比較器,用于排序元素優(yōu)先級(jí)
*/
private final Comparator<? super E> comparator;
}
可以看出PriorityQueue底層是基于數(shù)組實(shí)現(xiàn)的,使用Object[]數(shù)組存儲(chǔ)元素,并且定義了比較器comparator,用于排序元素的優(yōu)先級(jí)。
初始化
PriorityQueue常用的初始化方法有4個(gè):
- 無參構(gòu)造方法
- 指定容量大小的有參構(gòu)造方法
- 指定比較器的有參構(gòu)造方法
- 同時(shí)指定容量和比較器的有參構(gòu)造方法
/** * 無參構(gòu)造方法 */ PriorityQueue<Integer> blockingQueue1 = new PriorityQueue<>(); /** * 指定容量大小的構(gòu)造方法 */ PriorityQueue<Integer> blockingQueue2 = new PriorityQueue<>(10); /** * 指定比較器的有參構(gòu)造方法 */ PriorityQueue<Integer> blockingQueue3 = new PriorityQueue<>(Integer::compareTo); /** * 同時(shí)指定容量和比較器的有參構(gòu)造方法 */ PriorityQueue<Integer> blockingQueue4 = new PriorityQueue<>(10, Integer::compare);
再看一下對(duì)應(yīng)的源碼實(shí)現(xiàn):
/**
* 無參構(gòu)造方法
*/
public PriorityQueue() {
// 使用默認(rèn)容量大小11,不指定比較器
this(DEFAULT_INITIAL_CAPACITY, null);
}
/**
* 指定容量大小的構(gòu)造方法
*/
public PriorityQueue(int initialCapacity) {
this(initialCapacity, null);
}
/**
* 指定比較器的有參構(gòu)造方法
*/
public PriorityQueue(Comparator<? super E> comparator) {
this(DEFAULT_INITIAL_CAPACITY, comparator);
}
/**
* 同時(shí)指定容量和比較器的有參構(gòu)造方法
*/
public PriorityQueue(int initialCapacity, Comparator<? super E> comparator) {
if (initialCapacity < 1) {
throw new IllegalArgumentException();
}
this.queue = new Object[initialCapacity];
this.comparator = comparator;
}
可以看出PriorityQueue的無參構(gòu)造方法使用默認(rèn)的容量大小11,直接初始化數(shù)組,并且沒有指定比較器。
放數(shù)據(jù)源碼
放數(shù)據(jù)的方法有2個(gè):
| 操作 | 拋出異常 | 返回特定值 |
|---|---|---|
| 放數(shù)據(jù) | add() | offer() |
offer方法源碼
先看一下offer()方法源碼,其他放數(shù)據(jù)方法邏輯也是大同小異,都是在鏈表尾部插入。 offer()方法在隊(duì)列滿的時(shí)候,會(huì)直接返回false,表示插入失敗。
/**
* offer方法入口
*
* @param e 元素
* @return 是否插入成功
*/
public boolean offer(E e) {
// 1. 判空,傳參不允許為null
if (e == null) {
throw new NullPointerException();
}
modCount++;
int i = size;
// 2. 當(dāng)數(shù)組滿的時(shí)候,執(zhí)行擴(kuò)容
if (i >= queue.length) {
grow(i + 1);
}
size = i + 1;
// 3. 如果是第一次插入,就直接把元素插入到數(shù)組頭部
if (i == 0) {
queue[0] = e;
} else {
// 4. 如果不是第一次插入,就找個(gè)合適的位置插入(需要保證插入后數(shù)組有序)
siftUp(i, e);
}
return true;
}
offer()方法邏輯也很簡(jiǎn)單,先判斷是否需要擴(kuò)容,如果需要擴(kuò)容先執(zhí)行擴(kuò)容邏輯,然后把元素插入到數(shù)組中。如果是第一次插入,就直接把元素插入到數(shù)組頭部。如果不是,就找個(gè)合適的位置插入,需要保證插入后數(shù)組仍是有序的。 再看一下擴(kuò)容的源碼:
/**
* 擴(kuò)容
*/
private void grow(int minCapacity) {
int oldCapacity = queue.length;
// 1. 如果原數(shù)組容量小于64,就執(zhí)行2倍擴(kuò)容,否則執(zhí)行1.5擴(kuò)容
int newCapacity = oldCapacity +
((oldCapacity < 64) ? (oldCapacity + 2) : (oldCapacity >> 1));
// 2. 校驗(yàn)最大容量不能超過Integer最大值
if (newCapacity - MAX_ARRAY_SIZE > 0) {
newCapacity = hugeCapacity(minCapacity);
}
// 3. 直接擴(kuò)容后新數(shù)組賦值給原數(shù)組
queue = Arrays.copyOf(queue, newCapacity);
}
擴(kuò)容的源碼設(shè)計(jì)充滿了作者的巧思,在數(shù)組容量較小的時(shí)候,為了避免頻繁擴(kuò)容,就采用2倍擴(kuò)容法。在數(shù)組容量較大的時(shí)候,為了避免擴(kuò)容后浪費(fèi)空間,就采用1.5倍擴(kuò)容法。
PriorityQueue為了快速的插入和刪除,采用了最小堆,而不是直接使用有序數(shù)組,這樣既可以保證插入和刪除的時(shí)間復(fù)雜度都是O(logn),又能避免移動(dòng)過多元素。
最小堆的定義: 除葉子節(jié)點(diǎn)外,每個(gè)節(jié)點(diǎn)的值都小于等于左右子節(jié)點(diǎn)的值。
下面就是一個(gè)簡(jiǎn)單的最小堆和映射數(shù)組:

再看一下siftUp()方法源碼,是怎么保證插入元素,數(shù)組仍是有序的? 其實(shí)就是循環(huán)跟父節(jié)點(diǎn)比較元素大小,找個(gè)合適的位置插入。
// 把元素插入到合適的位置
private void siftUp(int k, E x) {
// 1. 如果初始化的時(shí)候,自定義了比較器,就使用自定義比較器的插入方法,否則使用默認(rèn)的。
if (comparator != null) {
siftUpUsingComparator(k, x);
} else {
siftUpComparable(k, x);
}
}
// 自定義比較器的插入方法
private void siftUpUsingComparator(int k, E x) {
while (k > 0) {
// 1. 找到父節(jié)點(diǎn)
int parent = (k - 1) >>> 1;
Object e = queue[parent];
// 2. 如果當(dāng)前節(jié)點(diǎn)元素比父節(jié)點(diǎn)的元素小,就把父節(jié)點(diǎn)元素向下移動(dòng)(給當(dāng)前元素騰出位置)
if (comparator.compare(x, (E) e) >= 0) {
break;
}
queue[k] = e;
k = parent;
}
// 3. 把當(dāng)前元素插入到父節(jié)點(diǎn)的位置
queue[k] = x;
}
// 默認(rèn)的插入方法
private void siftUpComparable(int k, E x) {
// 1. 使用默認(rèn)比較器
Comparable<? super E> key = (Comparable<? super E>) x;
while (k > 0) {
// 2. 找到父節(jié)點(diǎn)
int parent = (k - 1) >>> 1;
Object e = queue[parent];
// 3. 如果當(dāng)前節(jié)點(diǎn)元素比父節(jié)點(diǎn)的元素小,就把父節(jié)點(diǎn)元素向下移動(dòng)(給當(dāng)前元素騰出位置)
if (key.compareTo((E) e) >= 0) {
break;
}
queue[k] = e;
k = parent;
}
// 4. 把當(dāng)前元素插入到父節(jié)點(diǎn)的位置
queue[k] = key;
}
再看一下add()方法源碼:
add方法源碼
add()方法底層直接調(diào)用的是offer()方法,作用相同。
/**
* add方法入口
*
* @param e 元素
* @return 是否添加成功
*/
public boolean add(E e) {
return offer(e);
}
彈出數(shù)據(jù)源碼
彈出數(shù)據(jù)(取出數(shù)據(jù)并刪除)的方法有2個(gè):
| 操作 | 拋出異常 | 返回特定值 |
|---|---|---|
| 取數(shù)據(jù)(同時(shí)刪除數(shù)據(jù)) | remove() | poll() |
poll方法源碼
看一下poll()方法源碼,其他方取數(shù)據(jù)法邏輯大同小異,都是從數(shù)組頭部彈出元素。 poll()方法在彈出元素的時(shí)候,如果隊(duì)列為空,直接返回null,表示彈出失敗。
/**
* poll方法入口
*/
public E poll() {
// 1. 如果數(shù)組為空,返回null
if (size == 0) {
return null;
}
int s = --size;
modCount++;
// 2. 暫存數(shù)組頭節(jié)點(diǎn),最后返回
E result = (E) queue[0];
// 3. 暫存數(shù)組尾節(jié)點(diǎn),調(diào)整最小堆的時(shí)候,需要上移
E x = (E) queue[s];
// 4. 刪除尾節(jié)點(diǎn)
queue[s] = null;
// 5. 調(diào)整最小堆
if (s != 0) {
siftDown(0, x);
}
return result;
}
remove方法源碼
再看一下remove()方法源碼,如果隊(duì)列為空,remove()會(huì)拋出異常。
/**
* remove方法入口
*/
public E remove() {
// 1. 直接調(diào)用poll方法
E x = poll();
// 2. 如果取到數(shù)據(jù),直接返回,否則拋出異常
if (x != null) {
return x;
} else {
throw new NoSuchElementException();
}
}
查看數(shù)據(jù)源碼
再看一下查看數(shù)據(jù)源碼,查看數(shù)據(jù),并不刪除數(shù)據(jù)。
| 操作 | 拋出異常 | 返回特定值 |
|---|---|---|
| 查看數(shù)據(jù)(不刪除) | element() | peek() |
peek方法源碼
先看一下peek()方法源碼,如果數(shù)組為空,直接返回null。
/**
* peek方法入口
*/
public E peek() {
// 返回?cái)?shù)組頭節(jié)點(diǎn)
return (size == 0) ? null : (E) queue[0];
}
element方法源碼
再看一下element()方法源碼,如果隊(duì)列為空,則拋出異常。
/**
* element方法入口
*/
public E element() {
// 1. 調(diào)用peek方法查詢數(shù)據(jù)
E x = peek();
// 2. 如果查到數(shù)據(jù),直接返回
if (x != null) {
return x;
} else {
// 3. 如果沒找到,則拋出異常
throw new NoSuchElementException();
}
}
總結(jié)
這篇文章講解了PriorityQueue阻塞隊(duì)列的核心源碼,了解到PriorityQueue隊(duì)列具有以下特點(diǎn):
PriorityQueue實(shí)現(xiàn)了Queue接口,提供了兩組放數(shù)據(jù)和讀數(shù)據(jù)的方法,來滿足不同的場(chǎng)景。PriorityQueue底層基于數(shù)組實(shí)現(xiàn),按照最小堆存儲(chǔ),實(shí)現(xiàn)了高效的插入和刪除。PriorityQueue初始化的時(shí)候,可以指定數(shù)組長度和自定義比較器。PriorityQueue初始容量是11,當(dāng)數(shù)組容量小于64,采用2倍擴(kuò)容,否則采用1.5擴(kuò)容。PriorityQueue每次都是從數(shù)組頭節(jié)點(diǎn)取元素,取之后需要調(diào)整最小堆。
今天一起分析了PriorityQueue隊(duì)列的源碼,可以看到PriorityQueue的源碼非常簡(jiǎn)單,沒有什么神秘復(fù)雜的東西,下篇文章再一起接著分析其他的阻塞隊(duì)列源碼。
以上就是詳解Java中PriorityQueue的作用和源碼實(shí)現(xiàn)的詳細(xì)內(nèi)容,更多關(guān)于Java PriorityQueue的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
SpringBoot如何讀取war包jar包和Resource資源
這篇文章主要介紹了SpringBoot如何讀取war包jar包和Resource資源,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-01-01
spring mvc+localResizeIMG實(shí)現(xiàn)HTML5端圖片壓縮上傳
這篇文章主要為大家詳細(xì)介紹了使用spring mvc+localResizeIMG實(shí)現(xiàn)HTML5端圖片壓縮上傳,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2017-04-04
Java多線程從基礎(chǔ)到高級(jí)應(yīng)用示例小結(jié)
本章介紹了Java多線程編程的基礎(chǔ)知識(shí),包括線程的創(chuàng)建與管理、線程同步、線程通信和線程池,重點(diǎn)內(nèi)容包括線程的三種創(chuàng)建方式,通過本章的學(xué)習(xí),讀者掌握了Java多線程編程的基本技能,感興趣的朋友跟隨小編一起看看吧2026-01-01
如何使用IntelliJ IDEA的HTTP Client進(jìn)行接口驗(yàn)證
這篇文章主要介紹了如何使用IntelliJ IDEA的HTTP Client進(jìn)行接口驗(yàn)證,本文給大家分享最新完美解決方案,感興趣的朋友跟隨小編一起看看吧2024-06-06
SpringCloud Config統(tǒng)一配置中心問題分析解決與客戶端動(dòng)態(tài)刷新實(shí)現(xiàn)
springcloud config是一個(gè)解決分布式系統(tǒng)的配置管理方案。它包含了 client和server兩個(gè)部分,server端提供配置文件的存儲(chǔ)、以接口的形式將配置文件的內(nèi)容提供出去,client端通過接口獲取數(shù)據(jù)、并依據(jù)此數(shù)據(jù)初始化自己的應(yīng)用2022-10-10
JDK動(dòng)態(tài)代理與CGLib動(dòng)態(tài)代理的區(qū)別對(duì)比
今天小編就為大家分享一篇關(guān)于JDK動(dòng)態(tài)代理與CGLib動(dòng)態(tài)代理的區(qū)別對(duì)比,小編覺得內(nèi)容挺不錯(cuò)的,現(xiàn)在分享給大家,具有很好的參考價(jià)值,需要的朋友一起跟隨小編來看看吧2019-02-02

