Add Go codes to docs, including
the chapter of stack and queue, the chapter of tree.
This commit is contained in:
@@ -4,38 +4,41 @@
|
||||
|
||||
package chapter_tree
|
||||
|
||||
import "testing"
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestBinarySearchTree(t *testing.T) {
|
||||
nums := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}
|
||||
bst := NewBinarySearchTree(nums)
|
||||
t.Log("初始化的二叉树为:")
|
||||
fmt.Println("初始化的二叉树为:")
|
||||
bst.Print()
|
||||
|
||||
// 获取根结点
|
||||
node := bst.GetRoot()
|
||||
t.Log("二叉树的根结点为:", node.Val)
|
||||
fmt.Println("二叉树的根结点为:", node.Val)
|
||||
// 获取最小的结点
|
||||
node = bst.GetMin(bst.GetRoot())
|
||||
t.Log("二叉树的最小结点为:", node.Val)
|
||||
fmt.Println("二叉树的最小结点为:", node.Val)
|
||||
|
||||
// 查找结点
|
||||
node = bst.Search(5)
|
||||
t.Log("查找到的结点对象为", node, ",结点值 =", node.Val)
|
||||
fmt.Println("查找到的结点对象为", node, ",结点值 =", node.Val)
|
||||
|
||||
// 插入结点
|
||||
node = bst.Insert(16)
|
||||
t.Log("插入结点后 16 的二叉树为:")
|
||||
fmt.Println("插入结点后 16 的二叉树为:")
|
||||
bst.Print()
|
||||
|
||||
// 删除结点
|
||||
bst.Remove(1)
|
||||
t.Log("删除结点 1 后的二叉树为:")
|
||||
fmt.Println("删除结点 1 后的二叉树为:")
|
||||
bst.Print()
|
||||
bst.Remove(2)
|
||||
t.Log("删除结点 2 后的二叉树为:")
|
||||
fmt.Println("删除结点 2 后的二叉树为:")
|
||||
bst.Print()
|
||||
bst.Remove(4)
|
||||
t.Log("删除结点 4 后的二叉树为:")
|
||||
fmt.Println("删除结点 4 后的二叉树为:")
|
||||
bst.Print()
|
||||
}
|
||||
|
Reference in New Issue
Block a user