package main
import (
"fmt"
"testing"
)
func TestMinHeap(t *testing.T) {
h := NewMinHeap([]int{1, 4, 6, 7, 3})
fmt.Println(*h) // [1 3 6 7 4] // ok
h.Push(2)
fmt.Println(*h) // [1 3 2 7 4 6] // ok
fmt.Println(h.Pop()) // 1
fmt.Println(*h) // [2 3 6 7 4] // ok
}
func TestMaxHeap(t *testing.T) {
h := NewMaxHeap([]int{1, 2, 6, 5, 9})
fmt.Println(*h) // [9 5 6 1 2]
h.Push(8)
fmt.Println(*h) // [9 5 8 1 2 6]
fmt.Println(h.Pop())
fmt.Println(*h) // [8 5 6 1 2]
}
最后编辑: kuteng 文档更新时间: 2021-06-05 10:16 作者:kuteng