AVL平衡二叉树

type Iterator

  1. func (iterator *Iterator) Begin() // 将迭代器重置为初始状态,然后调用Next获取第一个元素
  2. func (iterator *Iterator) End() // 将迭代器移过最后一个元素,然后调用Prev获取最后一个元素
  3. // 将迭代器移动到第一个元素,如果容器中有第一个元素则返回true。
  4. // 如果First()返回true,则可以通过index()和value()检索第一个元素的索引和值。修改迭代器的状态
  5. func (iterator *Iterator) First() bool
  6. // 将迭代器移动到最后元素,如果容器中有第一个元素则返回true。
  7. // 如果Last()返回true,则可以通过index()和value()检索第一个元素的索引和值。修改迭代器的状态
  8. func (iterator *Iterator) Last() bool
  9. func (iterator *Iterator) Next() bool
  10. func (iterator *Iterator) Prev() bool
  11. func (iterator *Iterator) Index() int
  12. func (iterator *Iterator) Value() interface{}

type Node

  1. type Node struct {
  2. Key interface{}
  3. Value interface{}
  4. Parent *Node // Parent node
  5. Children [2]*Node // Children nodes
  6. // contains filtered or unexported fields
  7. }
  8. func (n *Node) Next() *Node
  9. func (n *Node) Prev() *Node
  10. func (n *Node) String() string

type Tree

  1. func NewWith(comparator utils.Comparator) *Tree
  2. // 实例化一个具有IntComparator的树,即键是int类型的
  3. func NewWithIntComparator() *Tree
  4. // 用StringComparator实例化一个树,即键的类型是string
  5. func NewWithStringComparator() *Tree
  6. // 找打大于或等于key节点的最小节点
  7. func (t *Tree) Ceiling(key interface{}) (floor *Node, found bool)
  8. func (t *Tree) Clear()
  9. func (t *Tree) Empty() bool
  10. // 找打小于或等于key节点的最小节点
  11. func (t *Tree) Floor(key interface{}) (floor *Node, found bool)
  12. func (tree *Tree) FromJSON(data []byte) error
  13. func (t *Tree) Get(key interface{}) (value interface{}, found bool)
  14. // 返回一个有序迭代器
  15. func (tree *Tree) Iterator() containers.ReverseIteratorWithKey
  16. func (t *Tree) Keys() []interface{}
  17. func (t *Tree) Left() *Node
  18. func (t *Tree) Put(key interface{}, value interface{})
  19. func (t *Tree) Remove(key interface{})
  20. func (t *Tree) Right() *Node
  21. func (t *Tree) Size() int
  22. func (t *Tree) String() string
  23. func (tree *Tree) ToJSON() ([]byte, error)
  24. func (t *Tree) Values() []interface{}

例子:

  1. package main
  2. import (
  3. "fmt"
  4. avl "github.com/emirpasic/gods/trees/avltree"
  5. )
  6. // AVLTreeExample to demonstrate basic usage of AVLTree
  7. func main() {
  8. tree := avl.NewWithIntComparator() // empty(keys are of type int)
  9. tree.Put(1, "x") // 1->x
  10. tree.Put(2, "b") // 1->x, 2->b (in order)
  11. tree.Put(1, "a") // 1->a, 2->b (in order, replacement)
  12. tree.Put(3, "c") // 1->a, 2->b, 3->c (in order)
  13. tree.Put(4, "d") // 1->a, 2->b, 3->c, 4->d (in order)
  14. tree.Put(5, "e") // 1->a, 2->b, 3->c, 4->d, 5->e (in order)
  15. tree.Put(6, "f") // 1->a, 2->b, 3->c, 4->d, 5->e, 6->f (in order)
  16. fmt.Println(tree)
  17. //
  18. // AVLTree
  19. // │ ┌── 6
  20. // │ ┌── 5
  21. // └── 4
  22. // │ ┌── 3
  23. // └── 2
  24. // └── 1
  25. _ = tree.Values() // []interface {}{"a", "b", "c", "d", "e", "f"} (in order)
  26. _ = tree.Keys() // []interface {}{1, 2, 3, 4, 5, 6} (in order)
  27. tree.Remove(2) // 1->a, 3->c, 4->d, 5->e, 6->f (in order)
  28. fmt.Println(tree)
  29. //
  30. // AVLTree
  31. // │ ┌── 6
  32. // │ ┌── 5
  33. // └── 4
  34. // └── 3
  35. // └── 1
  36. tree.Clear() // empty
  37. tree.Empty() // true
  38. tree.Size() // 0
  39. }