二叉堆与Java中的优先队列


之前在A*算法演示程序的编码过程中,发现javaScript并没有原生的优先队列,于是去Java中找到了PriorityQueue类,研究了一下源码。Java中的优先队列基于最小二叉堆实现。最小二叉堆具有两个性质:

  • 结构性:必须是一颗完全二叉树,树的插入从左到右,一棵完全二叉树的高度为小于log(N)的最大整数。

  • 堆序性:二叉树的父节点必须小于两个子节点。父节点下标为n,两个子节点下标为2n+1和2(n+1)。父节点的值总是小于子节点,两个子节点间大小关系不确定。

Java PriorityQueue特点:

  • 不接受null值

  • 不接受不可排序的值

  • 线程不安全

  • 容量没有上限

  • 插入和删除元素的时间复杂度为O(log(n))

首先来看下最关键的两个方法,siftDown(int k, E x)和siftUp(int k, E x)。这两个方法是插入、删除、排序和初始化操作的基础。两个方法的目的都是让节点k所在的子树符合二叉堆的性质,堆序性。区别在于,siftDown将节点k作为子树的父节点处理,siftUp则将节点k作为子树的子节点处理。

siftDown


private void siftDownComparable(int k, E x) {        Comparablekey = (Comparable)x;        int half = size >>> 1;        //k) c).compareTo((E) queue[right]) > 0)                c = queue[child = right];            //如果父节点小于子节点,当前子树已经满足条件            //不需要继续            if (key.compareTo((E) c) <= 0)                break;            //否则,父节点与子节点对调,继续上述步骤            queue[k] = c;            k = child;        }        queue[k] = key;    }~~~关于这部分代码,比较有意思的是迭代条件`k>>1`,我们分size为奇数和偶数两种情况来看:* 奇数:size为奇数,`half=(size-1)/2`,最后一个节点下标为`size-1`,父节点下标为`(size-1)/2-1`,所以`kkey = (Comparable) x;

while (k > 0) {

int parent = (k - 1) >>> 1;

Object e = queue[parent];

if (key.compareTo((E) e) >= 0)

break;

queue[k] = e;

k = parent;

}

queue[k] = key;

}

siftUp的代码很容易理解,递归的与父节点比较大小,然后根据结果决定是否需要对调位置。

heapify

接下来是PriorityQueue的初始化方法中最复杂的一个,从一个无需集合(Collection)生成一个优先队列。


private void initFromCollection(Collection c) {

initElementsFromCollection(c);

heapify();

}

initElementsFromCollection(c)方法很简单,将集合c中的数据放入队列,将集合c的大小赋予size属性。而heapify()方法就是将集合c转换为二叉堆的关键。


private void heapify() {

for (int i = (size >>> 1) - 1; i >= 0; i--)

siftDown(i, (E) queue[i]);

}

size>>>1的特殊性我们刚才已经讨论过了,这里看下heapify()是怎么做的,从(size>>>1)-1开始向上遍历,对每一个父节点,都要保证它所在的子树符合条件,那么整棵树都将符合条件。

add offer

向队列中添加元素,两者代码相同,之所以并存是因为PriorityQueue继承自Queue。


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方法进行调整。最坏的情况,新插入的元素比根节点的元素还要小,那么siftUp要执行到根节点所在的子树。

removeAt

删除队列中下标为i的元素。由于要保持完整二叉树的结构,删除元素后仍需要进行重新排序。


private E removeAt(int i) {

modCount++;

int s = --size;

//如果i就是最后一个节点的下标,直接删除

if (s == i)

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;

}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,732评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,496评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,264评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,807评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,806评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,675评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,029评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,683评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,704评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,666评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,773评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,413评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,016评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,204评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,083评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,503评论 2 343

推荐阅读更多精彩内容