欢迎光临
让我们一起努力

PriorityQueue原理分析——基于源码

在业务场景中,处理一个任务队列,可能需要依照某种优先级顺序,这时,Java中的PriorityQueue(优先队列)便可以派上用场。优先队列的原理与堆排序密不可分,可以参考我之前的一篇博客:

堆排序总结与实现

原理

PriorityQueue中维护一个Queue[]数组,在逻辑上把它理解成一个小根堆或大根堆,即一个完全二叉树,每一个三元组中父节点小于两个孩子结点(小根堆,如果是大于则是大根堆)。本博客以小根堆来进行说明,因为PriorityQueue默认实现小根堆,即小的数先出队,当然也可以自定义Comparator实现大根堆。

  • 入队:每次入队时,把新元素挂在最后,从下往上遍历调整成小根堆;
  • 出队:每次出队时,移除顶部元素,把最后的元素移到顶部,并从上往下遍历调整成小根堆。

出队

poll()方法如下:

public E poll() {
    if (size == 0)
        return null;
    int s = --size;
    modCount++;
    E result = (E) queue[0];
    E x = (E) queue[s];
    queue[s] = null;
    if (s != 0)
        siftDown(0, x);
    return result;
}

可以看到,队首元素 queue[0] 出队,队尾的元素 queue[s] 进入 siftDown(0, x) 方法进行堆调整。siftDown方法如下:

private void siftDown(int k, E x) {
    if (comparator != null)
        siftDownUsingComparator(k, x);
    else
        siftDownComparable(k, x);
}
//k为开始遍历的位置,x为需要插入的值
@SuppressWarnings("unchecked")
private void siftDownComparable(int k, E x) {
    Comparable<? super E> key = (Comparable<? super E>)x;
    int half = size >>> 1;        // loop while a non-leaf
    // 只需要遍历到数组的一半即可,保证遍历到最后一个三元组的父节点即可
    while (k < half) {
        int child = (k << 1) + 1; // assume left child is least
        Object c = queue[child];
        int right = child + 1;
        if (right < size &&
            ((Comparable<? super E>) c).compareTo((E) queue[right]) > 0)
            c = queue[child = right];//比较左右孩子结点,取最小的那个
        if (key.compareTo((E) c) <= 0)
            break;//找到了key应该放入的位置
        queue[k] = c;
        k = child;
    }
    queue[k] = key;
}

@SuppressWarnings("unchecked")
private void siftDownUsingComparator(int k, E x) {
    int half = size >>> 1;
    while (k < half) {
        int child = (k << 1) + 1;
        Object c = queue[child];
        int right = child + 1;
        if (right < size &&
            comparator.compare((E) c, (E) queue[right]) > 0)
            c = queue[child = right];
        if (comparator.compare(x, (E) c) <= 0)
            break;
        queue[k] = c;
        k = child;
    }
    queue[k] = x;
}

可以看到,这与堆排序中的堆调整如出一辙。

入队

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;
}

同样,其核心在于 siftUp(i, e) 方法。如下所示:

private void siftUp(int k, E x) {
    if (comparator != null)
        siftUpUsingComparator(k, x);
    else
        siftUpComparable(k, x);
}

@SuppressWarnings("unchecked")
private void siftUpComparable(int k, E x) {
    Comparable<? super E> key = (Comparable<? super E>) x;
    while (k > 0) {
        int parent = (k - 1) >>> 1;//结点父节点的下标
        Object e = queue[parent];
        if (key.compareTo((E) e) >= 0)
            break;//如果结点值大于父节点,则可以放置在该三元组下
        queue[k] = e;//向子节点赋值父节点的值,不用担心某些值被覆盖,因为初始k等于size
        k = parent;
    }
    queue[k] = key;//最后在待插入位置赋key的值
}

@SuppressWarnings("unchecked")
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;
}

此方法,是一个不断从父节点往子节点赋值的过程,直到找到适合放置插入结点值的位置。

移除

removeAt 方法如下所示:

private E removeAt(int i) {
    // assert i >= 0 && i < size;
    modCount++;
    int s = --size;
    if (s == i) // removed last element
        queue[i] = null;
    else {
        E moved = (E) queue[s];
        queue[s] = null;
        siftDown(i, moved);
        if (queue[i] == moved) {
            siftUp(i, moved);
            if (queue[i] != moved)
                return moved;
        }
    }
    return null;
}

移除下标为i的元素,相当于以 i 为根节点的完全二叉树的出队,于是执行 siftDown 方法调整最后一个元素 moved 的位置,即将该堆调整为小根堆。调整完之后,如果 moved 没有来到 i 的位置,说明 i 以上的堆结构一定符合规则;如果 moved 被调整到 i 位置,i上面的父节点有可能比 moved大,所以需要 siftUp(i, moved) 方法从 i 位置向上调整,调整为小根堆,完毕。

总结

其实不管是 siftUp 方法还是 siftDown 方法,都是利用了完全二叉树的性质,通过父节点与孩子结点之间的快速访问来实现的。

赞(0) 我这么幸苦,不打个赏再走?
本作品采用CC BY-NC-SA 4.0进行许可:小李子的blog » PriorityQueue原理分析——基于源码

评论 抢沙发

CAPTCHAis initialing...
  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏