/
binary_tree.go
95 lines (80 loc) · 1.54 KB
/
binary_tree.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
// Seriál "Programovací jazyk Go"
// https://www.root.cz/serialy/programovaci-jazyk-go/
//
// Padesátá devátá část
// Generické datové typy v jazyce Go?
// https://www.root.cz/clanky/genericke-datove-typy-v-jazyce-go/
//
// Repositář:
// https://github.com/tisnik/go-root/
//
// Seznam demonstračních příkladů z padesáté deváté části:
// https://github.com/tisnik/go-root/blob/master/article_59/README.md
package main
import (
"fmt"
)
type Item int
type Node struct {
Value Item
Left *Node
Right *Node
}
type BinaryTree struct {
Root *Node
}
func (bt *BinaryTree) Insert(value Item) {
node := &Node{value, nil, nil}
if bt.Root == nil {
bt.Root = node
} else {
insertNode(bt.Root, node)
}
}
func insertNode(node, newNode *Node) {
if newNode.Value < node.Value {
if node.Left == nil {
node.Left = newNode
} else {
insertNode(node.Left, newNode)
}
} else {
if node.Right == nil {
node.Right = newNode
} else {
insertNode(node.Right, newNode)
}
}
}
func printTree(node *Node, level int) {
if node != nil {
format := ""
for i := 0; i < level; i++ {
format += " "
}
format += "---[ "
level++
printTree(node.Left, level)
fmt.Printf(format+"%v\n", node.Value)
printTree(node.Right, level)
}
}
func main() {
var bt BinaryTree
bt.Insert(8)
bt.Insert(3)
bt.Insert(11)
bt.Insert(1)
bt.Insert(0)
bt.Insert(2)
bt.Insert(5)
bt.Insert(4)
bt.Insert(6)
bt.Insert(9)
bt.Insert(8)
bt.Insert(10)
bt.Insert(13)
bt.Insert(12)
bt.Insert(14)
printTree(bt.Root, 0)
}