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.
This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.
// File: avl_tree_test.go
// Created Time: 2023-01-08
// Author: Reanon (793584285@qq.com)
package chapter_tree
import (
"fmt"
"testing"
. "github.com/krahets/hello-algo/pkg"
)
func TestAVLTree ( t * testing . T ) {
/* 初始化空 AVL 樹 */
tree := newAVLTree ( )
/* 插入節點 */
// 請關注插入節點後, AVL 樹是如何保持平衡的
testInsert ( tree , 1 )
testInsert ( tree , 2 )
testInsert ( tree , 3 )
testInsert ( tree , 4 )
testInsert ( tree , 5 )
testInsert ( tree , 8 )
testInsert ( tree , 7 )
testInsert ( tree , 9 )
testInsert ( tree , 10 )
testInsert ( tree , 6 )
/* 插入重複節點 */
testInsert ( tree , 7 )
/* 刪除節點 */
// 請關注刪除節點後, AVL 樹是如何保持平衡的
testRemove ( tree , 8 ) // 刪除度為 0 的節點
testRemove ( tree , 5 ) // 刪除度為 1 的節點
testRemove ( tree , 4 ) // 刪除度為 2 的節點
/* 查詢節點 */
node := tree . search ( 7 )
fmt . Printf ( "\n查詢到的節點物件為 %#v ,節點值 = %d \n" , node , node . Val )
}
func testInsert ( tree * aVLTree , val int ) {
tree . insert ( val )
fmt . Printf ( "\n插入節點 %d 後, AVL 樹為 \n" , val )
PrintTree ( tree . root )
}
func testRemove ( tree * aVLTree , val int ) {
tree . remove ( val )
fmt . Printf ( "\n刪除節點 %d 後, AVL 樹為 \n" , val )
PrintTree ( tree . root )
}