1. 程式人生 > >java集合(2)——優先佇列的comparator

java集合(2)——優先佇列的comparator

  1. 優先佇列不允許空值,而且不支援non-comparable(不可比較)的物件,比如使用者自定義的類。
  2. 優先佇列要求使用Java Comparable和Comparator介面給物件排序,並且在排序時會按照優先順序處理其中的元素。
  3. 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();
        }

    });