mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-01 18:55:23 +08:00
Yudong Jin
2f561d3d51
* feat(ko): add the complete Korean online edition * Use UTF-8 for Python CI output on Windows * Remove print-book promotion from the Korean site * fix(ko): preserve emphasis before Korean particles
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
// File: array_binary_tree_test.go
|
|
// Created Time: 2023-07-24
|
|
// Author: Reanon ([email protected])
|
|
|
|
package chapter_tree
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
. "github.com/krahets/hello-algo/pkg"
|
|
)
|
|
|
|
func TestArrayBinaryTree(t *testing.T) {
|
|
// 이진 트리 초기화
|
|
// 여기서는 배열로부터 이진 트리를 바로 생성하는 함수를 사용한다
|
|
arr := []any{1, 2, 3, 4, nil, 6, 7, 8, 9, nil, nil, 12, nil, nil, 15}
|
|
root := SliceToTree(arr)
|
|
fmt.Println("\n이진 트리 초기화")
|
|
fmt.Println("이진 트리의 배열 표현:")
|
|
fmt.Println(arr)
|
|
fmt.Println("이진 트리의 연결 리스트 표현:")
|
|
PrintTree(root)
|
|
|
|
// 배열로 표현한 이진 트리 클래스
|
|
abt := newArrayBinaryTree(arr)
|
|
|
|
// 노드 접근
|
|
i := 1
|
|
l := abt.left(i)
|
|
r := abt.right(i)
|
|
p := abt.parent(i)
|
|
fmt.Println("\n현재 노드의 인덱스:", i, ", 값:", abt.val(i))
|
|
fmt.Println("왼쪽 자식 노드의 인덱스:", l, ", 값:", abt.val(l))
|
|
fmt.Println("오른쪽 자식 노드의 인덱스:", r, ", 값:", abt.val(r))
|
|
fmt.Println("부모 노드의 인덱스:", p, ", 값:", abt.val(p))
|
|
|
|
// 트리 순회
|
|
res := abt.levelOrder()
|
|
fmt.Println("\n레벨 순서 순회:", res)
|
|
res = abt.preOrder()
|
|
fmt.Println("전위 순회:", res)
|
|
res = abt.inOrder()
|
|
fmt.Println("중위 순회:", res)
|
|
res = abt.postOrder()
|
|
fmt.Println("후위 순회:", res)
|
|
}
|