Golang使用heap库实现简单的优先级队列

AmbitiousJun

已于 2022-08-16 16:49:36 修改

阅读量1.8k

点赞数 2

文章标签： golang 开发语言后端

于 2022-05-05 19:06:40 首次发布

本文链接：https://blog.csdn.net/C_L99/article/details/124595316

版权

使用heap库之前，首先要实现其定义的五个函数，分别是：

type Interface interface {
    // 堆的大小，通常返回的是切片的长度
    Len() int
    // 比较器，函数返回值为true时表示切片中i索引的元素比较“小”
	Less(i, j int) bool
    // 交换i和j两个坐标的元素
	Swap(i, j int)
    // 将元素x加入到堆中
	Push(x any)
    // 弹出堆顶元素
	Pop() any
}

一个大根堆的实现示例：

type bHp struct{ sort.IntSlice }

func (hp bHp) Less(i, j int) bool { return hp.IntSlice[i] > hp.IntSlice[j] }

func (hp bHp) Peek() interface{} {
	a := hp.IntSlice
	return a[0]
}

func (hp *bHp) Push(x interface{}) {
	hp.IntSlice = append(hp.IntSlice, x.(int))
}

func (hp *bHp) Pop() interface{} {
	a := hp.IntSlice
	x := a[len(a)-1]
	hp.IntSlice = a[:len(a)-1]
	return x
}

注意事项：

sort.IntSlice是golang标准库sort下定义的一个类型，它实现了上述提到的接口函数的前三个，分别是Len()、Less()、Swap()
结构体bHp直接使用sort.IntSlice类型作为值，可以将它默认实现的三个函数添加到bHp中（一般情况下，不需要再自己再去实现一次了）
heap库实现的是小根堆，但是可以由我们自己去定义何为“小”。这里要实现的是一个大根堆，所以我们需要重写Less()函数：只要i索引下的元素值比j索引下的元素值要大，就说明i元素在堆中更“小”。

对自定义的堆进行操作

使用heap库已经定义好的函数，将实例化好的自定义堆的指针传进去即可。

// 实例化自定义堆
pq := &bHp{}
// 插入值
heap.Push(pq, 8)
heap.Push(pq, 9)
// 获取堆顶元素但不删除
fmt.Printf("当前堆顶元素为：%d\n", pq.Peek())
heap.Push(pq, 7)
heap.Push(pq, 10)
heap.Push(pq, 5)
heap.Push(pq, 7)
// 获取堆的大小
fmt.Printf("当前堆大小为：%d\n", pq.Len())
// 将所有的元素弹出
for pq.Len() > 0 {
    fmt.Println(heap.Pop(pq))
}

为什么自定义的bHp结构没有实现Len函数却能直接调用？

这个问题在上文已经提到过，bHp结构体直接使用sort.IntSlice类型作为成员，就拥有了sort.IntSlice中实现的所有函数，可以直接调用。

Pop函数的实现中为什么取的是切片中最后一个值？

首先我们要知道，在heap库内部维护堆这个结构时，堆顶（即堆中“最小”的元素）是处于切片的第一个元素的

其次，我们在实际调用的时候不是直接bhp.Pop()这样调的，而是调用heap.Pop(bhp)，这就意味着Pop操作的时候程序不仅仅走了我们自定义的这段逻辑。查看heap.Pop()函数源码：

// Pop removes and returns the minimum element (according to Less) from the heap.
// The complexity is O(log n) where n = h.Len().
// Pop is equivalent to Remove(h, 0).
func Pop(h Interface) any {
	n := h.Len() - 1
	h.Swap(0, n)
	down(h, 0, n)
	return h.Pop()
}