binary tree
parent
d3e33651ac
commit
f23e579d4a
@ -0,0 +1,52 @@
|
||||
package binaryTree
|
||||
|
||||
type Comparable func (c1 interface{}, c2 interface{}) bool
|
||||
|
||||
type BinaryTree struct {
|
||||
node interface{}
|
||||
left *BinaryTree
|
||||
right *BinaryTree
|
||||
lessFun Comparable
|
||||
}
|
||||
|
||||
func New(compareFun Comparable) *BinaryTree {
|
||||
tree := &BinaryTree{}
|
||||
tree.node = nil
|
||||
tree.lessFun = compareFun
|
||||
return tree
|
||||
}
|
||||
|
||||
func (tree *BinaryTree) Search(value interface{}) *BinaryTree {
|
||||
if tree.node == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
if tree.node == value {
|
||||
return tree
|
||||
} else {
|
||||
if tree.lessFun(value, tree.node) == true {
|
||||
t := tree.left.Search(value)
|
||||
return t
|
||||
} else {
|
||||
t := tree.right.Search(value)
|
||||
return t
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (tree *BinaryTree) Insert(value interface{}) {
|
||||
if tree.node == nil {
|
||||
tree.node = value
|
||||
tree.right = New(tree.lessFun)
|
||||
tree.left = New(tree.lessFun)
|
||||
return
|
||||
} else {
|
||||
if tree.lessFun(value, tree.node) == true {
|
||||
tree.left.Insert(value)
|
||||
} else {
|
||||
tree.right.Insert(value)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -0,0 +1,30 @@
|
||||
package binaryTree
|
||||
|
||||
import "testing"
|
||||
|
||||
func compare(x interface{}, y interface{}) bool {
|
||||
if x.(int) < y.(int) {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func Test_binaryTree(t *testing.T) {
|
||||
tree := New(compare)
|
||||
|
||||
tree.Insert(1)
|
||||
tree.Insert(2)
|
||||
tree.Insert(3)
|
||||
|
||||
findTree := tree.Search(2)
|
||||
if findTree.node != 2 {
|
||||
t.Error("[Error] Search error")
|
||||
}
|
||||
|
||||
findNilTree := tree.Search(100)
|
||||
|
||||
if findNilTree != nil {
|
||||
t.Error("[Error] 2. Search erro")
|
||||
}
|
||||
}
|
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue