golang基于heap库实现简易优先队列-木庄网络博客

本文摘自网络，作者，侵删。

优先队列在C++/Java等语言中已经存在于标准库中，而go语言中却没有。今天用了下，留个纪念吧

package main
import (
    "container/heap"
    "fmt"
)

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{}) {
    *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
}

func TestIntHeap(t *testing.T) {
    pq := &IntHeap{}
    heap.Init(pq)
    for i := 1; i < 10; i++ {
        heap.Push(pq, i)
    }

    for len(*pq) > 0 {
        x := heap.Pop(pq)
        fmt.Println(x)
    }
}

/*
输出结果
1
2
3
4
5
6
7
8
9
*/

本文来自：简书

感谢作者：克罗地亚催眠曲

查看原文：golang基于heap库实现简易优先队列

相关阅读 >>

编写可测试的 Go 代码

关于面试

Go语言接口

手撸Golang 仿spring ioc/aop 之6 扫码1

Go语言反射的使用

Golang 架构设计原则里氏替换原则

Golang panic可以捕获标准错误吗

Go 语言环境安装