堆、堆排序与优先队列

堆的一个非常典型的应用就是优先队列,在说堆之前我们先说下,什么是优先队列?

普通的队列我们知道,就是由入队时间的顺序来决定出队的顺序,先进先出后进后出。这种规则满足我们生活中大多数场景,比如超市结账地铁进站等等,而优先队列则是出队的顺序和入队的顺序无关,而和优先级有关,比如医院排队就诊,有120送来的急诊患者肯定优先级最高要第一时间就诊。而在计算机中优先队列则被大量的使用,系统要同时执行多个任务,在执行时是将CPU的执行周期划成了时间片,在每个时间片里只能执行一个任务,执行哪一个任务就是取决于任务的优先级,系统会动态的选择当前优先级最高的任务来执行。所以此时就用到了优先队列,将所有需要执行的任务入列,由优先队列来选择优先级最高的任务。

可能此时有人会问,如果将所有任务按优先级进行一次排序不就可以吗?可现实是,当我们执行完一个任务之后,此时可能会再来好几个不同优先级的任务,而且刚刚拍好序的任务的优先级可能也会改变(原本很紧急,突然变得不紧急了),如果每次选择任务前都对任务进行一次排序,这样的耗时是很巨大的。

此时我们知道了优先队列的必要性,那么实现优先队列为什么要用堆呢?我们看下图


图1.png

如果我们用普通的数组,那么入队时很简单直接随便放入数组,只需要O(1)的时间,然而在出队时要挑选最大的那个优先级,则需要O(n)的时间。同样如果是有序的数组,入队时需要维护这个数组的有序性,找到此优先级合适的位置,需要O(n)的时间,出队就很简单只需要拿出数组第一个就好了。而如果使用堆的话,则可以非常好的平衡入队与出队的时间复杂度。使用堆可以将入队与出队的时间复杂度都变成lg(n)级别,虽然使用堆在入队上是慢于普通数组,在出队上是慢于有序数组的,可是平均来讲,使用堆来实现优先队列的时间复杂度大大低于我们使用数组来实现。

堆的定义

堆是一种特殊的数据结构,最为经典的实现是二叉堆,它是一棵完全二叉树。二叉树长这样
图2.png

这个二叉树特点就是树里的任何一个节点都不大于它的父节点,并且他是一个完全二叉树。所谓完全二叉树就是,除了最后一层以外,其他的层的节点必须是最大值。也就是说这一层没有装满,你就不能开下一层。而且最后一层剩余的节点,必须集中在左侧(如下图)。这样堆我们也称为最大堆,因为树顶的那个元素总是最大的那个值。反之同样也有最小堆

图3.png

堆的实现

既然堆是一个树形结构,我们可以像其他普通的树形结构一样,创建节点的对象,这个对象有两个指针,指向它的左孩子和右孩子。但对于堆而言我们可以用数组来存储一个堆。我可以用编号来标记每一个节点。

图4.png

我们可以看出,节点的序号乘以2就是它左孩子的节点,节点序号乘以2加1就是右孩子的节点。所以我们可以归纳出,在第一个节点用1来标记的情况下,节点 i 它的父节点为 i/2,左孩子节点为i*2,右孩子i*2+1。
Ps:从1开始标记是比较常用的实现方式,你也可以从0开始,相应的表达式也有改变。节点 i 的父节点为 (i-1)/2 , 左孩子为2*i+1 、右孩子2*i+2。

上浮操作 shift up

当我们需要向一个最大堆添加一条新的数据时,此时我们的堆变成了这样。

图5.png

此时,由于新数据的加入已经不符合最大堆的定义了。所以我们需要对新加入的数据进行shift up操作,将它放到它应该在的位置。shift up操作时我们将新加入的数据与它的父节点进行比较。如果比它的父节点大,则交换二者。
图6.png

并且不断的重复这个操作,将它与父节点比较,直到它不大于父节点,或者它没有父节点(到顶了)。
图7.png

此时我们就完成了 对新加入元素的shift up操作。
伪代码如下

  void shiftUp(int i) {
        while(i>1 && array[i/2]<array[i]) {
          swap(array[i/2],array[i]);
          i /=2;
        }
  }

下沉操作 shift down

当我们从堆中(也就是优先队列中)取出一个元素时。我们是将堆顶的元素弹出。(只能从堆顶取出)


图8.png

此时这个堆没有顶了,那么该怎么办呢?我们只需要把这个堆最后一个元素放到堆顶就可以了。


图9.png

图10.png

此时这个堆已经不符合最大堆的性质。为了保持这个性质,我们需要将堆顶的元素调整到它应该在的位置。也就是对它进行shift Down操作。在shift down时,因为它有左右孩子两个节点,所以我们需要将左右两个孩子节点进行比较,在得到较大的节点之后,再与它进行比较,如果它的子节点大,则将二者交换。并且不断的重复这样的操作,直到它没有叶子节点或者大于叶子节点停止

图11.png

图12.png

此时我们就完成了弹出一个元素之后的shift down操作。

伪代码如下

void  shiftDown(int i) {
  while(2*i <= array.count){
    int j = 2*i;//需要交换的目标索引,初始为左孩子的
    if(j+1 <= array.count && array[j+1]>array[j]) {//需要判断此节点有没有右孩子
      j += 1;
    }
    if(array[i] >= array[j]) {//此时子节点没有大于父节点直接跳出
      break;
    }
    swap(array[i] , array[j]);
    i = j;
  }
}

堆排序

堆排序(heap sort)是利用堆这种数据结构,来进行排序的一种算法。

首先将一个数组抽象成一个堆。这个过程,我们称之为heapify

图13.png

之后我们找到这个堆中第一个非叶子节点,这个节点的位置始终是数组的数量除以2,也就是索引5位置的27,从这个节点开始,对每一个非叶子的节点,,进行shift down操作。

27比它的子节点51要小,所以交换二者。


图14.png

之后对索引4位置的11,以及索引3位置的25,继续进行shift down。


图15.png
图16.png

接下来我们看索引2位置的20。首先呢,我们需要将20与它两个子节点中较大的51交换。
图17.png

此时,还没有完,shift down操作是直到它没有叶子节点或者大于叶子节点时才停止。
所以我们需要继续将20与它的子节点27进行比较。

图18.png

索引2才算操作完成了。
最后继续对索引1位置的14,以此类推进行shiftdown。整个堆就操作完成。使用的时候每次取出堆顶的元素,整个数组就是排序完成的了。
图19.png

如果觉得作者对哪里的理解有偏差,希望不吝赐教,留言指正。谢谢支持~

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

推荐阅读更多精彩内容