當前位置：首頁 > 網絡編程 > JavaScript > 詳細頁面

Java優先隊列(PriorityQueue)重寫compare操作

時間：2020-10-08來源：www.ship-models.net作者：電腦系統城

we can custom min heap or max heap by override the method compare.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38 package myapp.kit.quickstart.utils;

import java.util.Comparator;
import java.util.Queue;

/**
* priority queue (heap) demo.
*
* @author huangdingsheng
* @version 1.0, 2020/5/8
*/
public class PriorityQueue {

public static void main(String[] args) {

// min heap, process custom data struct
Queue<Node> minHeap = new java.util.PriorityQueue<>(new Comparator<Node>(){
@Override
public int compare(Node o1, Node o2) {
return o1.v - o2.v;
}
});

// do sth...
minHeap.add(new Node(1, 2));
minHeap.add(new Node(2, 9));
}

// custom data struct
static class Node {
int k;
int v;
public Node(int k, int v) {
this.k = k;
this.v = v;
}
}
}

補充知識：Java中PriorityQueue的排序，堆排序

PrioriryQueue是Queue接口的一個隊列實現類，但它的排序并不是典型的隊列式先進先出（FIFO）的方式。

PriorityQueue的排序方式分為兩種，一種是自然排序，這是按照加入元素的大小從小到大排序的。第二種是定制排序，是使用comparator類來重寫compare(Object o1,Object o2)方法來實現定制排序的。但是這些都不是關鍵，關鍵在于PriorityQueue的排序不是普通的排序，而是堆排序，這有什么不同呢？來看下面一段代碼：

1
2
3
4
5
6
7
8
9
10
11
12
13 import java.util.PriorityQueue;
public class PriorityQueueTest3
{
public static void main(String[] args)
{
PriorityQueue pq = new PriorityQueue();
pq.offer(6);
pq.offer(-3);
pq.offer(0);
pq.offer(9);
System.out.println(pq);
}
}

輸出結果：[-3,6,0,9]

不是說是按從小到大來排序的嗎？怎么沒排序？

原因是堆排序只會保證第一個元素也就是根節點的元素是當前優先隊列里最?。ɑ蛘咦畲螅┑脑?，而且每一次變化之后，比如offer()或者poll()之后，都會進行堆重排，所以如果想要按從小到大的順序取出元素，那么需要用一個for循環，如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 import java.util.PriorityQueue;
public class PriorityQueueTest3
{
public static void main(String[] args)
{
PriorityQueue pq = new PriorityQueue();
pq.offer(6);
pq.offer(-3);
pq.offer(0);
pq.offer(9);
int len = pq.size();//這里之所以取出.size()大小，因為每一次poll()之后size大小都會變化，所以不能作為for循環的判斷條件
for(int i=0;i<len;i++){
System.out.print(pq.poll()+" ");
}
System.out.println();
}
}