Merge pull request #17 from wlwanpan/binary_tree_min_max
Add min and max for binary treepull/20/head
commit
5840a8e776
@ -1,52 +1,60 @@
|
||||
package binaryTree
|
||||
|
||||
type Comparable func (c1 interface{}, c2 interface{}) bool
|
||||
type Comparable func(c1 interface{}, c2 interface{}) bool
|
||||
|
||||
type BinaryTree struct {
|
||||
node interface{}
|
||||
left *BinaryTree
|
||||
right *BinaryTree
|
||||
lessFun Comparable
|
||||
node interface{}
|
||||
left *BinaryTree
|
||||
right *BinaryTree
|
||||
lessFun Comparable
|
||||
}
|
||||
|
||||
func New(compareFun Comparable) *BinaryTree {
|
||||
tree := &BinaryTree{}
|
||||
tree.node = nil
|
||||
tree.lessFun = compareFun
|
||||
return tree
|
||||
return &BinaryTree{
|
||||
node: nil,
|
||||
lessFun: compareFun,
|
||||
}
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
}
|
||||
if tree.node == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
if tree.node == value {
|
||||
return tree
|
||||
}
|
||||
if tree.lessFun(value, tree.node) {
|
||||
return tree.left.Search(value)
|
||||
} else {
|
||||
return tree.right.Search(value)
|
||||
}
|
||||
}
|
||||
|
||||
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)
|
||||
}
|
||||
}
|
||||
if tree.node == nil {
|
||||
tree.node = value
|
||||
tree.right = New(tree.lessFun)
|
||||
tree.left = New(tree.lessFun)
|
||||
return
|
||||
}
|
||||
if tree.lessFun(value, tree.node) {
|
||||
tree.left.Insert(value)
|
||||
} else {
|
||||
tree.right.Insert(value)
|
||||
}
|
||||
}
|
||||
|
||||
func (tree *BinaryTree) Max() interface{} {
|
||||
if tree.node == nil || tree.right.node == nil {
|
||||
return tree.node
|
||||
}
|
||||
return tree.right.Max()
|
||||
}
|
||||
|
||||
func (tree *BinaryTree) Min() interface{} {
|
||||
if tree.node == nil || tree.left.node == nil {
|
||||
return tree.node
|
||||
}
|
||||
return tree.left.Min()
|
||||
}
|
||||
|
Loading…
Reference in New Issue