From 3dcdd1c72dafdae44ab5e7bfdbc06918a3adff93 Mon Sep 17 00:00:00 2001
From: reanon <793584285@qq.com>
Date: Fri, 13 Jan 2023 17:37:24 +0800
Subject: [PATCH] fix(heap): add go codes

---
 codes/go/chapter_heap/heap.go | 40 +++++++++++++++++++++++++++++++++++
 docs/chapter_heap/heap.md     | 36 ++++++++++++++++++++++++++++++-
 2 files changed, 75 insertions(+), 1 deletion(-)

diff --git a/codes/go/chapter_heap/heap.go b/codes/go/chapter_heap/heap.go
index 5f846ced..886ec20e 100644
--- a/codes/go/chapter_heap/heap.go
+++ b/codes/go/chapter_heap/heap.go
@@ -4,6 +4,11 @@
 
 package chapter_heap
 
+import (
+	"container/heap"
+	"fmt"
+)
+
 // Go 语言中可以通过实现 heap.Interface 来构建整数大顶堆
 // 实现 heap.Interface 需要同时实现 sort.Interface
 type intHeap []any
@@ -43,3 +48,38 @@ func (h *intHeap) Swap(i, j int) {
 func (h *intHeap) Top() any {
 	return (*h)[0]
 }
+
+/* Driver Code */
+func main() {
+	/* 初始化堆 */
+	// 初始化大顶堆
+	maxHeap := &intHeap{}
+	heap.Init(maxHeap)
+	/* 元素入堆 */
+	// 调用 heap.Interface 的方法，来添加元素
+	heap.Push(maxHeap, 1)
+	heap.Push(maxHeap, 3)
+	heap.Push(maxHeap, 2)
+	heap.Push(maxHeap, 4)
+	heap.Push(maxHeap, 5)
+
+	/* 获取堆顶元素 */
+	top := maxHeap.Top()
+	fmt.Printf("堆顶元素为 %d\n", top)
+
+	/* 堆顶元素出堆 */
+	// 调用 heap.Interface 的方法，来移除元素
+	heap.Pop(maxHeap)
+	heap.Pop(maxHeap)
+	heap.Pop(maxHeap)
+	heap.Pop(maxHeap)
+	heap.Pop(maxHeap)
+
+	/* 获取堆大小 */
+	size := len(*maxHeap)
+	fmt.Printf("堆元素数量为 %d\n", size)
+
+	/* 判断堆是否为空 */
+	isEmpty := len(*maxHeap) == 0
+	fmt.Printf("堆是否为空 %t\n", isEmpty)
+}
diff --git a/docs/chapter_heap/heap.md b/docs/chapter_heap/heap.md
index 91f7670b..92499325 100644
--- a/docs/chapter_heap/heap.md
+++ b/docs/chapter_heap/heap.md
@@ -136,6 +136,41 @@ comments: true
     func (h *intHeap) Top() any {
         return (*h)[0]
     }
+
+    /* Driver Code */
+    func TestHeap(t *testing.T) {
+        /* 初始化堆 */
+        // 初始化大顶堆
+        maxHeap := &intHeap{}
+        heap.Init(maxHeap)
+        /* 元素入堆 */
+        // 调用 heap.Interface 的方法，来添加元素
+        heap.Push(maxHeap, 1)
+        heap.Push(maxHeap, 3)
+        heap.Push(maxHeap, 2)
+        heap.Push(maxHeap, 4)
+        heap.Push(maxHeap, 5)
+
+        /* 获取堆顶元素 */
+        top := maxHeap.Top()
+        fmt.Printf("堆顶元素为 %d\n", top)
+
+        /* 堆顶元素出堆 */
+        // 调用 heap.Interface 的方法，来移除元素
+        heap.Pop(maxHeap)
+        heap.Pop(maxHeap)
+        heap.Pop(maxHeap)
+        heap.Pop(maxHeap)
+        heap.Pop(maxHeap)
+
+        /* 获取堆大小 */
+        size := len(*maxHeap)
+        fmt.Printf("堆元素数量为 %d\n", size)
+
+        /* 判断堆是否为空 */
+        isEmpty := len(*maxHeap) == 0
+        fmt.Printf("堆是否为空 %t\n", isEmpty)
+    }
     ```
 
 === "JavaScript"
@@ -317,7 +352,6 @@ comments: true
     func (h *maxHeap) peek() any {
         return h.data[0]
     }
-
     ```
 
 === "JavaScript"