0

0

Golangcontainer/heap实现堆数据结构示例

P粉602998670

P粉602998670

发布时间:2025-09-12 11:08:01

|

506人浏览过

|

来源于php中文网

原创

答案:Go语言通过container/heap包提供堆操作,需实现heap.Interface并使用heap.Init、heap.Push等函数初始化和维护堆结构。

golangcontainer/heap实现堆数据结构示例

Go语言标准库中的container/heap包提供了一个堆(优先队列)的接口实现,但不直接提供完整的堆类型。你需要先实现heap.Interface,然后使用heap.Initheap.Push

type IntHeap []int

// 实现 sort.Interface
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] }

// 实现 heap.Interface 的 Push 和 Pop
func (h *IntHeap) Push(x interface{}) {
    *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
}

使用堆的基本操作

初始化堆,并进行插入、删除顶部元素等操作:

package main

import (
    "container/heap"
    "fmt"
)

func main() {
    h := &IntHeap{3, 1, 4, 1, 5}
    
    // 初始化堆
    heap.Init(h)
    
    // 插入元素
    heap.Push(h, 2)
    heap.Push(h, 6)
    
    // 弹出最小元素(最小堆)
    for h.Len() > 0 {
        min := heap.Pop(h).(int)
        fmt.Print(min, " ") // 输出: 1 1 2 3 4 5 6
    }
}

自定义结构体堆示例(如任务优先级)

更常见的场景是基于结构体字段排序,比如按优先级排序的任务:

立即学习go语言免费学习笔记(深入)”;

大米商城damishop
大米商城damishop

damishop介绍 大米外贸商城系统 简称damishop 完全开源版,只需做一种语言一键开启全球133中语言自动翻译功能,价格实现自动汇率转换,集成微信支付宝 paypal以及国外主流支付方式,自带文章博客系统,首创支持可视化编辑。 软件架构 基于MVC+语言包模式,增加控制台,API导入产品方便对接其他系统(带json示例数据)。 使用要求

下载
type Task struct {
    ID       int
    Priority int
}

type TaskHeap []*Task

func (th TaskHeap) Len() int { return len(th) }

func (th TaskHeap) Less(i, j int) bool {
    return th[i].Priority < th[j].Priority // 优先级数值越小,越优先
}

func (th TaskHeap) Swap(i, j int) {
    th[i], th[j] = th[j], th[i]
}

func (th *TaskHeap) Push(x interface{}) {
    *th = append(*th, x.(*Task))
}

func (th *TaskHeap) Pop() interface{} {
    old := *th
    n := len(old)
    task := old[n-1]
    *th = old[0 : n-1]
    return task
}

使用方式类似:

tasks := &TaskHeap{
    {ID: 1, Priority: 3},
    {ID: 2, Priority: 1},
    {ID: 3, Priority: 2},
}
heap.Init(tasks)
heap.Push(tasks, &Task{ID: 4, Priority: 0})

for tasks.Len() > 0 {
    task := heap.Pop(tasks).(*Task)
    fmt.Printf("Task ID: %d, Priority: %d\n", task.ID, task.Priority)
}
// 输出按优先级升序

基本上就这些。只要实现了heap.Interface(包含sort.Interface + Push/Pop),就能用container/heap管理你的数据结构。注意PushPop操作的是指针接收者,且必须配合heap包函数调用,不能直接调用。

相关专题

更多
treenode的用法
treenode的用法

​在计算机编程领域,TreeNode是一种常见的数据结构,通常用于构建树形结构。在不同的编程语言中,TreeNode可能有不同的实现方式和用法,通常用于表示树的节点信息。更多关于treenode相关问题详情请看本专题下面的文章。php中文网欢迎大家前来学习。

534

2023.12.01

C++ 高效算法与数据结构
C++ 高效算法与数据结构

本专题讲解 C++ 中常用算法与数据结构的实现与优化,涵盖排序算法(快速排序、归并排序)、查找算法、图算法、动态规划、贪心算法等,并结合实际案例分析如何选择最优算法来提高程序效率。通过深入理解数据结构(链表、树、堆、哈希表等),帮助开发者提升 在复杂应用中的算法设计与性能优化能力。

17

2025.12.22

深入理解算法:高效算法与数据结构专题
深入理解算法:高效算法与数据结构专题

本专题专注于算法与数据结构的核心概念,适合想深入理解并提升编程能力的开发者。专题内容包括常见数据结构的实现与应用,如数组、链表、栈、队列、哈希表、树、图等;以及高效的排序算法、搜索算法、动态规划等经典算法。通过详细的讲解与复杂度分析,帮助开发者不仅能熟练运用这些基础知识,还能在实际编程中优化性能,提高代码的执行效率。本专题适合准备面试的开发者,也适合希望提高算法思维的编程爱好者。

15

2026.01.06

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

1019

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

63

2025.10.17

php8.4实现接口限流的教程
php8.4实现接口限流的教程

PHP8.4本身不内置限流功能,需借助Redis(令牌桶)或Swoole(漏桶)实现;文件锁因I/O瓶颈、无跨机共享、秒级精度等缺陷不适用高并发场景。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

412

2025.12.29

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

389

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

572

2023.08.10

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

2

2026.01.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
golang socket 编程
golang socket 编程

共2课时 | 0.1万人学习

nginx浅谈
nginx浅谈

共15课时 | 0.8万人学习

golang和swoole核心底层分析
golang和swoole核心底层分析

共3课时 | 0.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号