Add min and max for binary tree
parent
cc26ad7dce
commit
16fdc8debc
@ -1,52 +1,60 @@
|
|||||||
package binaryTree
|
package binaryTree
|
||||||
|
|
||||||
type Comparable func (c1 interface{}, c2 interface{}) bool
|
type Comparable func(c1 interface{}, c2 interface{}) bool
|
||||||
|
|
||||||
type BinaryTree struct {
|
type BinaryTree struct {
|
||||||
node interface{}
|
node interface{}
|
||||||
left *BinaryTree
|
left *BinaryTree
|
||||||
right *BinaryTree
|
right *BinaryTree
|
||||||
lessFun Comparable
|
lessFun Comparable
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(compareFun Comparable) *BinaryTree {
|
func New(compareFun Comparable) *BinaryTree {
|
||||||
tree := &BinaryTree{}
|
return &BinaryTree{
|
||||||
tree.node = nil
|
node: nil,
|
||||||
tree.lessFun = compareFun
|
lessFun: compareFun,
|
||||||
return tree
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tree *BinaryTree) Search(value interface{}) *BinaryTree {
|
func (tree *BinaryTree) Search(value interface{}) *BinaryTree {
|
||||||
if tree.node == nil {
|
if tree.node == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if tree.node == value {
|
if tree.node == value {
|
||||||
return tree
|
return tree
|
||||||
} else {
|
}
|
||||||
if tree.lessFun(value, tree.node) == true {
|
if tree.lessFun(value, tree.node) {
|
||||||
t := tree.left.Search(value)
|
return tree.left.Search(value)
|
||||||
return t
|
} else {
|
||||||
} else {
|
return tree.right.Search(value)
|
||||||
t := tree.right.Search(value)
|
}
|
||||||
return t
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tree *BinaryTree) Insert(value interface{}) {
|
func (tree *BinaryTree) Insert(value interface{}) {
|
||||||
if tree.node == nil {
|
if tree.node == nil {
|
||||||
tree.node = value
|
tree.node = value
|
||||||
tree.right = New(tree.lessFun)
|
tree.right = New(tree.lessFun)
|
||||||
tree.left = New(tree.lessFun)
|
tree.left = New(tree.lessFun)
|
||||||
return
|
return
|
||||||
} else {
|
}
|
||||||
if tree.lessFun(value, tree.node) == true {
|
if tree.lessFun(value, tree.node) {
|
||||||
tree.left.Insert(value)
|
tree.left.Insert(value)
|
||||||
} else {
|
} else {
|
||||||
tree.right.Insert(value)
|
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