java集合(2)——優先佇列的comparator
阿新 • • 發佈:2019-02-07
- 優先佇列不允許空值,而且不支援non-comparable(不可比較)的物件,比如使用者自定義的類。
- 優先佇列要求使用Java Comparable和Comparator介面給物件排序,並且在排序時會按照優先順序處理其中的元素。
PriorityQueue是非執行緒安全的,所以Java提供了PriorityBlockingQueue(實現BlockingQueue介面)用於Java多執行緒環境。
原始碼檢視
1.首先先看add
public boolean add(E e) {
return offer(e);
}
2.進入offer
public boolean offer(E e) {
if (e == 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;
}
3.檢視siftup
private void siftUp(int k, E x) {
if (comparator != null)
siftUpUsingComparator(k, x);
else
siftUpComparable(k, x);
}
4.檢視siftUpUsingComparator(k, x)
private void siftUpUsingComparator(int k, E x) {
while (k > 0 ) {
int parent = (k - 1) >>> 1;
Object e = queue[parent];
if (comparator.compare(x, (E) e) >= 0)
break;
queue[k] = e;
k = parent;
}
queue[k] = x;
}
5.
在上一個方法中,k是最後一個元素的索引,所以e代表最後一個元素,x代表插入元素。所以在重寫的比較器comparator(o1,o2)中,o1代表要插入的元素,o2代表最後一個元素。所以o1-o2是小根堆,o2-o1是大根堆。
private static PriorityQueue<Person> pq = new PriorityQueue<Person>(new Comparator<Person>() {
public int compare(Person o1, Person o2) {
// TODO Auto-generated method stub
return o1.getId()-o2.getId();
}
});