快捷導(dǎo)航

Java優(yōu)先隊(duì)列的創(chuàng)建與使用詳解

更新時(shí)間：2024年08月01日 09:21:44 作者：蒲公英的歲月

這篇文章主要介紹了Java優(yōu)先隊(duì)列的創(chuàng)建與使用,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

Java優(yōu)先隊(duì)列的創(chuàng)建與使用

最近刷到了力扣的第347題.前K個(gè)高頻元素，由于之前一直用C++刷題，對(duì)于Java的語(yǔ)法還有些生疏，這道題目需要使用優(yōu)先隊(duì)列解題，一下子不知道怎么創(chuàng)建了。

首先在Java中優(yōu)先隊(duì)列是由PriorityQueue實(shí)現(xiàn)Queue的接口，可以對(duì)隊(duì)列中的元素按照大小順序進(jìn)行排序，對(duì)于隊(duì)列中的元素，可以是基本數(shù)據(jù)類型的包裝類，也可以是自定義的類。

1.基本數(shù)據(jù)類型作為隊(duì)列元素

我們以Integer，整數(shù)類型的包裝類為例，構(gòu)造一個(gè)優(yōu)先隊(duì)列

Queue<Integer>que = new PriorityQueue<>();
que.add(2);
que.add(1);
que.add(3);
//在這種默認(rèn)的構(gòu)造方式下，隊(duì)列中的元素是從小到大排序的
//隊(duì)列中元素順序依次為1,2,3

如果要構(gòu)造最大優(yōu)先隊(duì)列，可以使用lambda表達(dá)式

代碼如下所示：

//注意這里return x1-x2為最小優(yōu)先隊(duì)列，x2-x1為最大優(yōu)先隊(duì)列
Queue<Integer>que = new PriorityQueue<>((x1,x2)->{return x2-x1;});
que.add(2);
que.add(1);
que.add(3);
//此時(shí)隊(duì)列中的元素依次為3,2,1

2.自定義數(shù)據(jù)類型作為隊(duì)列元素

假設(shè)我們需要使用Map<Integer,Integer>來(lái)對(duì)出現(xiàn)過(guò)的整數(shù)統(tǒng)計(jì)出現(xiàn)過(guò)的次數(shù)，即按照Map的值進(jìn)行排序，應(yīng)該如何做呢？

看如下代碼：

Map<Integer,Integer>map = new HashMap<>();
//....整數(shù)出現(xiàn)次數(shù)進(jìn)行統(tǒng)計(jì)，假設(shè)已經(jīng)全部統(tǒng)計(jì)進(jìn)入map中
//用一個(gè)集合來(lái)獲取map中的所有鍵值對(duì)
Set<Map.Entry<Integer,Integer>> set = map.entrySet();
//創(chuàng)建優(yōu)先隊(duì)列 同樣使用lambda表達(dá)式，getValue()函數(shù)用于獲取map中的值
//x1-x2即按照升序，是最小優(yōu)先隊(duì)列
Queue<Map.Entry<Integer,Integer>>que = new PriorityQueue<>((x1,x2)->{return x1.getValue()-x2.getValue();})
//再把Set中的鍵值對(duì)加入que中 則元素按照值的升序排列在隊(duì)列que中
for(Map.Entry<Integer,Integer> en:set){
que.add(en); //或que.offer(en);
}

使用Map<Integer,Integer>類型創(chuàng)建優(yōu)先隊(duì)列的過(guò)程如上所示，如果是一些自定義的類，比如學(xué)生的學(xué)號(hào)，年齡，進(jìn)行排序，同樣可以這樣操作，關(guān)于PriorityQueue的方法可以查看java api手冊(cè)