GO语言实现 一 堆与优先队列

堆与优先队列

优先队列

之前我们讲过队列这种数据结构,队列的特点是先进先出,那什么是优先队列呢?一般来说,优先队列分为两种,一种是最大优先队列,一种是最小优先队列

我们以最大优先队列为例来讲解今天的内容,最大优先队列的进出队列顺序并不是按照先进先出的准则,而是最大值先出

譬如队列 a = [3,1,5,4,2,0,6,9,7],按照最大优先队列的出队列顺序,下一个值应该是 9,然后是 7

使用数组解决最大优先队列一般有两种方法:

  1. 将数组排序,每次取出第一个值即可
  2. 每次出队列都遍历数组,取出最大值

有序数组的入队时间复杂度为 O(n),出队时间复杂度为 O(1)

无序数组的入队时间复杂度为 O(1),出队时间复杂度为 O(n)

无论有序还是无序,每次出入队列的操作的时间复杂度都是 O(n)

接下来我们介绍一种数据结构,它能够在优先队列的操作中达到 O(logn)的时间复杂度

堆的本质是完全二叉树(也有多叉的堆,不在今天讨论范围内)。对比于普通的完全二叉树,堆(大顶堆)有如下特性:

  1. 根节点为整个堆的最大值
  2. 每个节点的值都不小于它的子节点

由于完全二叉树的特性,我们可以将堆使用数组存储起来,而不用担心浪费空间

我们假设使用 a数组表示一个大顶堆,则

  1. a[1]元素为整个堆的最大值(a[0]元素暂被省略)
  2. k节点的父节点为 k/2
  3. k节点的孩子节点为 2k和 2k+1

在建堆的过程中,我们会遇到一些违反以上规则的情况,我们将通过以下两个函数来维护堆

func swim(k int, datas *[]int){
   for k>1 && less((*datas)[k/2],(*datas)[k]){
       exch(&(*datas)[k/2], &(*datas)[k])
       k = k/2
   }
}
func sink(k int, datas []int){
    N:=len(datas)-2
    for 2*k<=N{
        j:=2*k
        // 选取 j和 j+1中的最大值,假设最大值为 j,如果 j小于 j+1,则 j++变成 j+1
        // j这个变量永远指向子元素中的最大值
        if j<N && less(datas[j], datas[j+1]){
            j++
        }
        // 如果父节点比子节点要大,说明不用下沉了
        if !less(datas[k], datas[j]){
            break
        }
        // 下沉
        exch(&datas[k], &datas[j])
        k = j
    }
}

第一个函数是 swim主要用于元素入队,我们通过 swim方法找到新入队元素应该在的位置

第二个函数是 sink主要用于元素出队,我们在提取最大值时,会将队列末尾的元素与最大值(也就是队首)交换,然后下沉该元素,找到他应该在的位置

整个建堆的过程如下:

  1. 在一个空堆中插入第一个元素,此时该堆是符合堆的性质的
  2. 我们插入一个新元素,然后使用 swim 函数调整该新元素的位置
  3. 循环进行第二步,直到所有元素均已插入,此时整个堆建立完毕

整个建堆的过程可以看成不断执行入队列的过程,由于每个元素只与其父元素比较,所以整个建堆过程的时间复杂度为 O(logn)

然后我们考虑出队列,出队的过程如下:

  1. 由于队列的第一个值为整个队列的最大值,所以将队列的最后一个元素与队列的第一个元素互换,然后取出该最大值
  2. 此时由于队列最后一个元素在队首,肯定是不符合堆的性质的,所以我们使用 sink下沉该元素,通过与子节点比较,找到该元素合适的位置

此时我们的最大元素已经被取出,而整个队列还维持这堆的性质。由于出队列下沉的过程中,每个元素只与该元素的子节点比较,所以整个出队列过程的时间复杂度也是 O(logn)

golang代码实现如下:

func main(){
    datas:=[]int{
        4,1,5,7,8,1,2,12,63,64,23,46,76,7,123,12535,3,
    }
    buildHead(datas)
}

func less(a int,b int) bool {
    return a<b
}

func exch(pa *int, pb *int){
    temp:=*pa
    *pa = *pb
    *pb = temp
}

func swim(k int, datas *[]int){
    for k>1 && less((*datas)[k/2],(*datas)[k]){
        exch(&(*datas)[k/2], &(*datas)[k])
        k = k/2
    }
}

func insert(x int, datas []int) []int{
    datas = append(datas,x)
    //fmt.Println(datas)
    swim(len(datas)-1, &datas)
    return datas
}

func sink(k int, datas []int){
    N:=len(datas)-2
    for 2*k<=N{
        j:=2*k
        // 选取 j和 j+1中的最大值,假设最大值为 j,如果 j小于 j+1,则 j++变成 j+1
        // j这个变量永远指向子元素中的最大值
        if j<N && less(datas[j], datas[j+1]){
            j++
        }
        // 如果父节点比子节点要大,说明不用下沉了
        if !less(datas[k], datas[j]){
            break
        }
        // 下沉
        exch(&datas[k], &datas[j])
        k = j
    }
}

func delMax(datas *[]int)int{
    max:=(*datas)[1]
    exch(&(*datas)[1], &(*datas)[len((*datas))-1])
    sink(1, (*datas))
    *datas = (*datas)[:len((*datas))-1]
    return max
}

func buildHead(datas []int){
    ans:=make([]int,1)
    for i:=0; i<len(datas); i++{
        ans = insert(datas[i], ans)
    }
    cache:=len(ans)
    for i:=0; i<cache-1; i++{
        fmt.Println(delMax(&ans))
    }
    return ans
}

除了大顶堆,还有小顶堆,也就是队首元素为整个队列的最小值,整体逻辑基本不变,只是在元素比较时需要转变符号

堆排序

有了以上优先队列和堆的基础,堆排序的概念也很容易理解了。

我们可以将整个待排序的数组维护成一个堆,然后不断的使用删除最大值的操作,此时整个堆的输出即为降序排列。如果维护小顶堆,则输出为升序队列

如果我们仔细观察,会发现其实我们不需要对队列中的每一个元素执行 sink操作,如果此时整个数组已经确定,我们只需要对前 n/2个元素执行 sink操作即可(因为每个 sink操作会使该元素与子节点比较,所以对第n/2个元素执行 sink操作时,第 n个元素也被加入比较中)


参考文献:
Dynamic Connectivity - 普林斯顿大学 | Coursera

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

推荐阅读更多精彩内容