在 Go 中如何使用 heap 包？

Question

我一直在尝试在 Go 中使用 Heap 包，但我不确定如何初始化它。

package main
import "container/heap"

type PriorityMessage struct {
    Priority int
    Message string
}

func priorityQueue() {
    //WOULD THIS not initialize the heap?
    h := heap.Init(h PriorityMessage)

}

我一直试图在网上找到其他人如何初始化他们的堆的示例，并且他们似乎每次都创建自己的 Go 堆包版本。从 heap 包调用 heap.Init(h Interface) 函数不起作用吗？

Answer 1

这里有你应该首先实现的 heap.Interface。

type Interface interface {
    sort.Interface
    Push(x interface{}) // add x as element Len()
    Pop() interface{}   // remove and return element Len() - 1.
}

这意味着您应该为 PriorityMessage 结构提供必要的方法。将结构实例传递到 heap.Init(&pm) 之后。

您可以在 cmets 中链接的 godoc 中找到详细信息。

只是为了澄清混乱。 Go 是一种缺乏泛型的强类型语言。所以堆包的设计方式是与类型无关的。您可以为要实现的所有类型创建自己的实现。任何类型都实现了 heap.Interface 可以被 heap 包使用。

Answer 2

  //https://cs.opensource.google/go/go/+/refs/tags/go1.16.6:src/container/heap/example_intheap_test.go 
    
    // This example demonstrates an integer heap built using the heap interface.
    //package heap_test
    
    import (
        "container/heap"
        "fmt"
    )
    
    // An IntHeap is a min-heap of ints.
    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
    }
    
    // This example inserts several ints into an IntHeap, checks the minimum,
    // and removes them in order of priority.
    func Example_intHeap() {
        h := &IntHeap{2, 1, 5}
        heap.Init(h)
        heap.Push(h, 3)
        fmt.Printf("minimum: %d\n", (*h)[0])
        for h.Len() > 0 {
            fmt.Printf("%d ", heap.Pop(h))
        }
        // Output:
        // minimum: 1
        // 1 2 3 5
    }