You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
hello-algo/codes/go/chapter_tree/binary_tree_test.go

42 lines
752 B

// File: binary_tree_test.go
// Created Time: 2022-11-25
// Author: Reanon (793584285@qq.com)
package chapter_tree
import (
"fmt"
"testing"
. "github.com/krahets/hello-algo/pkg"
)
func TestBinaryTree(t *testing.T) {
/* 初始化二叉树 */
// 初始化节点
n1 := NewTreeNode(1)
n2 := NewTreeNode(2)
n3 := NewTreeNode(3)
n4 := NewTreeNode(4)
n5 := NewTreeNode(5)
// 构建引用指向(即指针)
n1.Left = n2
n1.Right = n3
n2.Left = n4
n2.Right = n5
fmt.Println("初始化二叉树")
PrintTree(n1)
/* 插入与删除节点 */
// 插入节点
p := NewTreeNode(0)
n1.Left = p
p.Left = n2
fmt.Println("插入节点 P 后")
PrintTree(n1)
// 删除节点
n1.Left = n2
fmt.Println("删除节点 P 后")
PrintTree(n1)
}