mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-08 13:36:06 +00:00
Yudong Jin
772183705e
* Add Russian docs site baseline * Add Russian localized codebase * Polish Russian code wording * Update ru code translation. * Update code translation and chapter covers. * Fix pythontutor extraction. * Add README and landing page. * placeholder of profiles * Use figures of English version * Remove chapter paperbook
55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
// 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)
|
|
}
|