go container/heap包浅析

2022-02-21 01:32:26 阅读：192 来源： 互联网

标签：container IntHeap int Pop IntSlice interface func go 浅析

go container/heap包提供了堆的实现。

更详细的定义见下面的链接

Go语言标准库文档中文版 | Go语言中文网 | Golang中文社区 | Golang中国 (studygolang.com)

这里主要讲下怎么使用这个包来实现最小堆和最大堆

首先要定义一个满足下面这个接口的类型

type Interface interface {
    sort.Interface
    Push(x interface{}) // 向末尾添加元素
    Pop() interface{}   // 从末尾删除元素
}

这意味着，有两种定义数据类型的方式，即是否复用sort包里的数据类型。

1.复用的情况，定义如下

type hp struct{ sort.IntSlice }
func (h *hp) Push(v interface{}) { h.IntSlice = append(h.IntSlice, v.(int)) }
func (h *hp) Pop() interface{}   { a := h.IntSlice; v := a[len(a)-1]; h.IntSlice = a[:len(a)-1]; return v }

2.不复用的情况

type IntHeap []int
func (h IntHeap) Len() int           { return len(h) }
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] }
func (h IntHeap) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }
func (h *IntHeap) Push(x interface{}) {
    // Push and Pop use pointer receivers because they modify the slice's length,
    // not just its contents.
    *h = append(*h, x.(int))
}
func (h *IntHeap) Pop() interface{} {
    old := *h
    n := len(old)
    x := old[n-1]
    *h = old[0 : n-1]
    return x
}

唯一需要注意的是Pop的实现，它是要删除数组末尾的元素，而非开头的元素。因为这个Pop是让heap包的Pop去调用的。heap包里Pop实际的逻辑是，先把堆的根节点值和数组末尾节点的值进行交换，然后再删除末尾的节点（调用我们实现的Pop），调整根节点。

另外，这个是最小堆的实现，如果想实现最大堆，有几种办法。

把Less的实现修改下，即返回结果取反。那么实际实现的就是最大堆了
把存入堆的值取负，那么这时候，最小堆其实就是绝对值的最大堆，取数据的时候再取负就行了。
最正规的办法，似乎是使用sort.Reverse

标签：container,IntHeap,int,Pop,IntSlice,interface,func,go,浅析
来源： https://www.cnblogs.com/dongjl/p/15917223.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

go container/heap包浅析