【Go - 小顶堆/大顶堆】

在 Go 语言中,标准库 container/heap 提供了堆(Heap)的实现。可以使用 container/heap 包来实现自己的大顶堆或小顶堆。

小顶堆示例

以下是一个使用 container/heap 包实现的小顶堆示例:

go 复制代码
package main

import (
	"container/heap"
	"fmt"
)

type IntHeap []int

func (h IntHeap) Len() int {
	return len(h)
}
func (h IntHeap) Less(i int, j int) bool {
	return h[i] < h[j]
}
func (h IntHeap) Swap(i int, j int) {
	h[i], h[j] = h[j], h[i]
}
func (h *IntHeap) Pop() interface{} {
	old := *h
	n := len(old)
	x := old[n-1]
	*h = old[0:n-1]
	return x
}
func (h *IntHeap) Push(x interface{}) {
	*h = append(*h, x.(int))
}

func main() {
	h := &IntHeap{2, 1, 5}
	heap.Init(h)
	heap.Push(h, 3)
	m := (*h)[0]
	fmt.Printf("minimum: %d\n", m)
	
	m = heap.Pop(h).(int)
	fmt.Printf("minimum %d\n", m)

	m = heap.Pop(h).(int)
	fmt.Printf("minimum %d\n", m)

}

注意📢 :

  • 涉及到空间变化的,Push,Pop 传参是指针类型,其余的是引用类型。
  • Push,Pop的参数为interface{},在内部再进行类型转换

大顶堆,则只需将Less函数改下,小于改为大于,其余代码一致。

go 复制代码
func (h IntHeap) Less(i int, j int) bool {
	return h[i] > h[j]
}
相关推荐
程序员编程指南11 分钟前
Qt 与 WebService 交互开发
c语言·开发语言·c++·qt·交互
*愿风载尘*12 分钟前
ksql连接数据库免输入密码交互
数据库·后端
赵英英俊15 分钟前
Python day26
开发语言·python
你怎么知道我是队长15 分钟前
python---eval函数
开发语言·javascript·python
溟洵18 分钟前
Qt 窗口 工具栏QToolBar、状态栏StatusBar
开发语言·前端·数据库·c++·后端·qt
ppo9229 分钟前
MCP简单应用:使用SpringAI + Cline + DeepSeek实现AI创建文件并写入内容
人工智能·后端
创码小奇客37 分钟前
Talos 使用全攻略:从基础到高阶,常见问题一网打尽
java·后端·架构
Re27537 分钟前
ThreadLocal 入门:搞懂线程私有变量
后端
心.c1 小时前
JavaScript单线程实现异步
开发语言·前端·javascript·ecmascript
midsummer_woo1 小时前
基于spring boot的纺织品企业财务管理系统(源码+论文)
java·spring boot·后端