咨询干货

了解优化推广技术相关资讯

Go语言中怎么实现一个小根堆,Java程序员表示看懵了

2019-05-25来自网络作者: 沙茶敏碎碎念

堆是我们一个常用的数据结构,堆是一个完全二叉树,下图是一个小根堆,小根堆的意思是对于树里面的所有节点,都是父节点小于任何一个子节点,今天我们来看一下Go中是怎么实现堆?


Go语言中怎么实现一个小根堆,Java程序员表示看懵了



下面是Go语言中实现一个堆的代码,对于每一个堆,我们都需要实现5个方法,这点感觉比Java的优先队列要复杂得多了,分别是Len,用来计算容器得长度,Less返回两个元素得大小关系,Swap,用来交换两个数据。这几个实际上都是用来实现sort接口的。接下来,我们需要实现Push跟Pop方法,对于一个以前不是写Go语言的人,可能觉得这个实现其实很冗余,Push方法我们要做的就是往数据结构的最后面插入一个元素,而Pop则是弹出一个元素,弹出元素则是简单的把最后一个元素取出来。小根堆不是最前面的元素才是最小的么?为什么是取最后一个元素?


Go语言中怎么实现一个小根堆,Java程序员表示看懵了



我们看一下go语言中,heap的源码,堆需要你实现Push跟Pop接口,因为继承了sort接口,所以又要实现上面3个比较方法。


Go语言中怎么实现一个小根堆,Java程序员表示看懵了



下面则是sort的接口,要求你实现长度,小于跟交换。


Go语言中怎么实现一个小根堆,Java程序员表示看懵了



接下来则是堆里面的push操作,先是调用了你实现的往末尾添加一个元素的接口,然后执行up操作,维护小根堆。up操作的目的是为了保证小根堆里面每一个结点都比子节点小。


Go语言中怎么实现一个小根堆,Java程序员表示看懵了



刚刚我们提出一个疑问,小根堆不是最前面的元素才最小么?这里堆里面的Pop方法是先把最小的元素放到最后面,然后再来维护这个长度减一的小根堆。所以实际上你上面执行方法的时候,最后一个元素才是最小的!

这里不由感叹,还是Java的PriorityQueue封装的好,使用起来更加方便。虽然很多人特别推崇Go,但我觉得每种语言都有各自的优点缺点,都说Go写起来很方便,这不,还是Java大法好。


声明:本站发布的内容以原创、转载、分享网络内容为主,如有侵权,请联系电话:021-51697771-8029,邮箱:mj@cndns.com,我们将会在第一时间删除。文章观点不代表本站立场,如需处理请联系我们。