答案是container/heap包需實(shí)現(xiàn)heap.Interface接口，通過定義Len、Less、Swap、Push、Pop方法構(gòu)建堆結(jié)構(gòu)，如用IntHeap實(shí)現(xiàn)最小堆或最大堆，并可擴(kuò)展至TaskHeap按優(yōu)先級(jí)處理任務(wù)。

在Go語言中，container/heap 是一個(gè)標(biāo)準(zhǔn)庫包，提供了堆（優(yōu)先隊(duì)列）的接口和操作方法。它本身沒有直接提供一個(gè)現(xiàn)成的堆結(jié)構(gòu)，而是要求你實(shí)現(xiàn)一個(gè)滿足 heap.Interface 接口的類型，然后通過 heap.Init、heap.Push、heap.Pop 等函數(shù)來操作這個(gè)堆。

1. 實(shí)現(xiàn) heap.Interface 接口

要使用 container/heap，你需要定義一個(gè)類型（通常是切片），并實(shí)現(xiàn)以下五個(gè)方法：

• Len() int：返回元素個(gè)數(shù)
• Less(i, j int) bool：定義堆的排序規(guī)則（最小堆或最大堆）
• Swap(i, j int)：交換兩個(gè)元素
• Push(x interface{})：向堆中添加元素
• Pop() interface{}：從堆中移除并返回元素（通常是堆頂）

2. 創(chuàng)建一個(gè)最小堆示例

下面是一個(gè)整數(shù)最小堆的完整實(shí)現(xiàn)：

package main

import (
    "container/heap"
    "fmt"
)

// 定義一個(gè)整數(shù)切片類型
type IntHeap []int

// 實(shí)現(xiàn) Len 方法
func (h IntHeap) Len() int           { return len(h) }

// Less 決定是小頂堆（<）還是大頂堆（>）
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] } // 最小堆

// Swap 交換元素
func (h IntHeap) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }

// Push 添加元素（注意：接收者是指針）
func (h *IntHeap) Push(x interface{}) {
    *h = append(*h, x.(int))
}

// Pop 移除并返回堆頂元素
func (h *IntHeap) Pop() interface{} {
    old := *h
    n := len(old)
    x := old[n-1]
    *h = old[0 : n-1]
    return x
}

func main() {
    h := &IntHeap{3, 1, 4, 1, 5}
    heap.Init(h)           // 初始化為堆
    heap.Push(h, 2)        // 插入元素
    fmt.Printf("最小值: %d\n", (*h)[0])
    
    for h.Len() > 0 {
        min := heap.Pop(h).(int)
        fmt.Print(min, " ")
    }
    // 輸出: 1 1 2 3 4 5
}

3. 創(chuàng)建一個(gè)最大堆

只需修改 Less 方法的比較方向：

立即學(xué)習(xí)“go語言免費(fèi)學(xué)習(xí)筆記（深入）”；

ViiTor實(shí)時(shí)翻譯

AI實(shí)時(shí)多語言翻譯專家！強(qiáng)大的語音識(shí)別、AR翻譯功能。

116

查看詳情

func (h IntHeap) Less(i, j int) bool { return h[i] > h[j] } // 最大堆

4. 處理復(fù)雜數(shù)據(jù)結(jié)構(gòu)（如任務(wù)優(yōu)先級(jí)）

實(shí)際開發(fā)中，堆常用于處理帶優(yōu)先級(jí)的對(duì)象。例如，按優(yōu)先級(jí)調(diào)度任務(wù)：

type Task struct {
    ID       int
    Priority int
}

type TaskHeap []*Task

func (h TaskHeap) Len() int            { return len(h) }
func (h TaskHeap) Less(i, j int) bool  { return h[i].Priority < h[j].Priority } // 優(yōu)先級(jí)小的先出（最小堆）
func (h TaskHeap) Swap(i, j int)       { h[i], h[j] = h[j], h[i] }

func (h *TaskHeap) Push(x interface{}) { *h = append(*h, x.(*Task)) }
func (h *TaskHeap) Pop() interface{} {
    old := *h
    n := len(old)
    task := old[n-1]
    *h = old[0 : n-1]
    return task
}

5. 常用操作總結(jié)

• heap.Init(h)：將已有的切片初始化為堆（O(n)）
• heap.Push(h, x)：插入元素（O(log n)）
• heap.Pop(h)：彈出堆頂（O(log n)）
• heap.Remove(h, i)：刪除指定索引的元素
• heap.Fix(h, i)：當(dāng)某個(gè)元素改變后，重新調(diào)整堆

基本上就這些。只要實(shí)現(xiàn)了 heap.Interface，就能利用 container/heap 提供的高效堆操作。關(guān)鍵在于正確實(shí)現(xiàn) Less 方法以控制堆序，以及 Push/Pop 使用指針接收者。這種設(shè)計(jì)靈活但需要手動(dòng)包裝，適合構(gòu)建優(yōu)先隊(duì)列、Dijkstra 算法、合并 K 個(gè)有序鏈表等場景。

以上就是Golang如何使用container/heap實(shí)現(xiàn)堆的詳細(xì)內(nèi)容，更多請(qǐng)關(guān)注php中文網(wǎng)其它相關(guān)文章！