gb_trees (stdlib v7.3)
View SourceGeneral balanced trees.
This module provides Prof. Arne Andersson's General Balanced Trees. These have no storage overhead compared to unbalanced binary trees, and their performance is better than AVL trees.
This module considers two keys as different if and only if they do not compare
equal (==).
Data Structure
Trees and iterators are built using opaque data structures that should not be pattern-matched from outside this module.
There is no attempt to balance trees after deletions. As deletions do not increase the height of a tree, this should be OK.
The original balance condition h(T) <= ceil(c * log(|T|)) has been changed to
the similar (but not quite equivalent) condition 2 ^ h(T) <= |T| ^ c. This
should also be OK.
See Also
Summary
Functions
Rebalances Tree1.
Removes the node with key Key from Tree1, returning the new tree;
raises an exception if Key is not present.
Removes the node with key Key from Tree1 if present and returns the
resulting tree; otherwise, returns Tree1 unchanged.
Returns a new empty tree.
Inserts Key with value Value into Tree1 if not present, or
updates the value for Key to Value if present; returns the new
tree.
Turns an ordered list List of key-value tuples into a tree.
Retrieves the value stored with Key in Tree; raises an exception
if Key is not present.
Inserts Key with value Value into Tree1, returning the new
tree; raises an exception if Key is already present.
Returns true if Key is present in Tree; otherwise, returns
false.
Returns true if Tree is an empty tree; otherwise, returns false.
Returns an iterator that can be used for traversing the entries of
Tree; see next/1.
Returns an iterator that can be used for traversing the entries of Tree in
either ordered or reversed direction; see next/1.
Returns an iterator that can be used for traversing the entries of
Tree starting from Key; see next/1.
Returns an iterator that can be used for traversing the entries of Tree in
either ordered or reversed direction starting from Key; see next/1.
Returns the keys in Tree as an ordered list.
Returns {Key2, Value}, where Key2 is the least key strictly greater than
Key1, Value is the value associated with this key.
Returns {Key, Value}, where Key is the largest key in Tree, and Value is
the value associated with this key.
Looks up Key in Tree and returns {value, Value} if found, or none if not present.
Maps function F(K, V1) -> V2 to all key-value pairs of tree Tree1,
returning a new tree Tree2 with the same set of keys as
Tree1 and the new set of values V2.
Returns {Key, Value, Iter2}, where Key is the next key referred to by
iterator Iter1, and Iter2 is the new iterator to be used for traversing the
remaining nodes, or the atom none if no nodes remain.
Returns the number of nodes in Tree.
Returns {Key2, Value}, where Key2 is the greatest key strictly less than
Key1, and Value is the value associated with this key.
Returns {Key, Value}, where Key is the smallest key in Tree, and Value
is the value associated with this key.
Returns a value Value from node with key Key and new Tree2
with that node removed.
Removes the node with key Key from Tree1 if present; otherwise, returns the tree unchanged.
Returns {Key, Value, Tree2}, where Key is the largest key in
Tree1, Value is the value associated with this key, and Tree2 is
this tree with the corresponding node deleted.
Returns {Key, Value, Tree2}, where Key is the smallest key in Tree1,
Value is the value associated with that key, and Tree2 is the tree
with the corresponding node removed.
Converts a tree into an ordered list of key-value tuples.
Updates Key to value Value in Tree1 and returns the new tree.
Returns the values in Tree as an ordered list, sorted by their
corresponding keys.
Types
Functions
Rebalances Tree1.
Note that this is rarely necessary, but can be motivated when many nodes have been deleted from the tree without further insertions. Rebalancing can then be forced to minimize lookup times, as deletion does not rebalance the tree.
Examples
1> Tree1 = gb_trees:from_orddict([{I,2*I} || I <- lists:seq(1, 100)]).
2> Delete = fun gb_trees:delete/2.
3> Tree2 = lists:foldl(Delete, Tree1, lists:seq(1, 50)).
4> gb_sets:size(Tree2).
50
5> Tree3 = gb_trees:balance(Tree2).
Removes the node with key Key from Tree1, returning the new tree;
raises an exception if Key is not present.
Examples
1> Tree1 = gb_trees:from_orddict([{a,1},{b,2}]).
2> Tree2 = gb_trees:delete(a, Tree1).
3> gb_trees:to_list(Tree2).
[{b,2}]
Removes the node with key Key from Tree1 if present and returns the
resulting tree; otherwise, returns Tree1 unchanged.
Examples
1> Tree1 = gb_trees:from_orddict([{a,1},{b,2}]).
2> Tree2 = gb_trees:delete_any(a, Tree1).
3> gb_trees:to_list(Tree2).
[{b,2}]
4> Tree3 = gb_trees:delete_any(z, Tree2).
5> Tree2 == Tree3.
true
Returns a new empty tree.
Examples
1> gb_trees:to_list(gb_trees:empty()).
[]
Inserts Key with value Value into Tree1 if not present, or
updates the value for Key to Value if present; returns the new
tree.
Examples
1> Tree1 = gb_trees:from_orddict([{a,1},{b,2}]).
2> Tree2 = gb_trees:enter(c, 10, Tree1).
3> Tree3 = gb_trees:enter(a, 100, Tree2).
4> gb_trees:to_list(Tree3).
[{a,100},{b,2},{c,10}]
-spec from_orddict(List) -> Tree when List :: [{Key, Value}], Tree :: tree(Key, Value).
Turns an ordered list List of key-value tuples into a tree.
The list must not contain duplicate keys.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2}]).
2> gb_trees:to_list(Tree).
[{a,1},{b,2}]
-spec get(Key, Tree) -> Value when Tree :: tree(Key, Value).
Retrieves the value stored with Key in Tree; raises an exception
if Key is not present.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2}]).
2> gb_trees:get(b, Tree).
2
Inserts Key with value Value into Tree1, returning the new
tree; raises an exception if Key is already present.
Examples
1> Tree1 = gb_trees:from_orddict([{a,1},{b,2}]).
2> Tree2 = gb_trees:insert(c, 10, Tree1).
3> gb_trees:to_list(Tree2).
[{a,1},{b,2},{c,10}]
Returns true if Key is present in Tree; otherwise, returns
false.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:is_defined(a, Tree).
true
3> gb_trees:is_defined(x, Tree).
false
Returns true if Tree is an empty tree; otherwise, returns false.
Examples
1> gb_trees:is_empty(gb_trees:empty()).
true
2> gb_trees:is_empty(gb_trees:from_orddict([{a,99}])).
false
Returns an iterator that can be used for traversing the entries of
Tree; see next/1.
Equivalent to iterator(Tree, ordered).
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> Iter0 = gb_trees:iterator(Tree).
3> {a,1,Iter1} = gb_trees:next(Iter0).
-spec iterator(Tree, Order) -> Iter when Tree :: tree(Key, Value), Iter :: iter(Key, Value), Order :: ordered | reversed.
Returns an iterator that can be used for traversing the entries of Tree in
either ordered or reversed direction; see next/1.
The implementation of this is very efficient; traversing the whole tree using
next/1 is only slightly slower than getting the list of all
elements using to_list/1 and traversing that. The main advantage of the
iterator approach is that it does not require the complete list of all elements
to be built in memory at one time.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> Iter0 = gb_trees:iterator(Tree, ordered).
3> {a,1,Iter1} = gb_trees:next(Iter0).
4> RevIter0 = gb_trees:iterator(Tree, reversed).
5> {c,3,RevIter1} = gb_trees:next(RevIter0).
Returns an iterator that can be used for traversing the entries of
Tree starting from Key; see next/1.
The difference as compared to the iterator returned by iterator/1 is
that the iterator starts with the first key greater than or equal to Key.
Equivalent to iterator_from(Key, Tree, ordered).
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3},{d,4}]).
2> Iter0 = gb_trees:iterator_from(aa, Tree).
3> {b,2,Iter1} = gb_trees:next(Iter0).
4> {c,3,Iter2} = gb_trees:next(Iter1).
-spec iterator_from(Key, Tree, Order) -> Iter when Tree :: tree(Key, Value), Iter :: iter(Key, Value), Order :: ordered | reversed.
Returns an iterator that can be used for traversing the entries of Tree in
either ordered or reversed direction starting from Key; see next/1.
The difference as compared to the iterator returned by iterator/2 is
that the iterator starts with the first key next to or equal to Key.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3},{d,4}]).
2> Iter0 = gb_trees:iterator_from(aa, Tree, ordered).
3> {b,2,Iter1} = gb_trees:next(Iter0).
4> RevIter0 = gb_trees:iterator_from(c, Tree, reversed).
5> {c,3,RevIter1} = gb_trees:next(RevIter0).
6> {b,2,RevIter2} = gb_trees:next(RevIter1).
Returns the keys in Tree as an ordered list.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:keys(Tree).
[a,b,c]
3> gb_trees:keys(gb_trees:empty()).
[]
-spec larger(Key1, Tree) -> none | {Key2, Value} when Key1 :: Key, Key2 :: Key, Tree :: tree(Key, Value).
Returns {Key2, Value}, where Key2 is the least key strictly greater than
Key1, Value is the value associated with this key.
Returns none if no such pair exists.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:larger(c, Tree).
none
3> gb_trees:larger(bb, Tree).
{c,3}
4> gb_trees:larger(a, Tree).
{b,2}
-spec largest(Tree) -> {Key, Value} when Tree :: tree(Key, Value).
Returns {Key, Value}, where Key is the largest key in Tree, and Value is
the value associated with this key.
Assumes that the tree is not empty.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:largest(Tree).
{c,3}
-spec lookup(Key, Tree) -> none | {value, Value} when Tree :: tree(Key, Value).
Looks up Key in Tree and returns {value, Value} if found, or none if not present.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:lookup(a, Tree).
{value,1}
3> gb_trees:lookup(z, Tree).
none
-spec map(Function, Tree1) -> Tree2 when Function :: fun((K :: Key, V1 :: Value1) -> V2 :: Value2), Tree1 :: tree(Key, Value1), Tree2 :: tree(Key, Value2).
Maps function F(K, V1) -> V2 to all key-value pairs of tree Tree1,
returning a new tree Tree2 with the same set of keys as
Tree1 and the new set of values V2.
Examples
1> Tree0 = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> Tree1 = gb_trees:map(fun(_, V) -> 2 * V end, Tree0).
3> gb_trees:to_list(Tree1).
[{a,2},{b,4},{c,6}]
-spec next(Iter1) -> none | {Key, Value, Iter2} when Iter1 :: iter(Key, Value), Iter2 :: iter(Key, Value).
Returns {Key, Value, Iter2}, where Key is the next key referred to by
iterator Iter1, and Iter2 is the new iterator to be used for traversing the
remaining nodes, or the atom none if no nodes remain.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> Iter0 = gb_trees:iterator(Tree).
3> {a,1,Iter1} = gb_trees:next(Iter0).
4> {b,2,Iter2} = gb_trees:next(Iter1).
5> {c,3,Iter3} = gb_trees:next(Iter2).
6> none = gb_trees:next(Iter3).
-spec size(Tree) -> non_neg_integer() when Tree :: tree().
Returns the number of nodes in Tree.
Examples
1> gb_trees:size(gb_trees:empty()).
0
2> gb_trees:size(gb_trees:from_orddict([{a,1},{b,2}])).
2
-spec smaller(Key1, Tree) -> none | {Key2, Value} when Key1 :: Key, Key2 :: Key, Tree :: tree(Key, Value).
Returns {Key2, Value}, where Key2 is the greatest key strictly less than
Key1, and Value is the value associated with this key.
Returns none if no such pair exists.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:smaller(c, Tree).
{b,2}
3> gb_trees:smaller(bb, Tree).
{b,2}
4> gb_trees:smaller(a, Tree).
none
-spec smallest(Tree) -> {Key, Value} when Tree :: tree(Key, Value).
Returns {Key, Value}, where Key is the smallest key in Tree, and Value
is the value associated with this key.
Assumes that the tree is not empty.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:smallest(Tree).
{a,1}
-spec take(Key, Tree1) -> {Value, Tree2} when Tree1 :: tree(Key, _), Tree2 :: tree(Key, _), Key :: term(), Value :: term().
Returns a value Value from node with key Key and new Tree2
with that node removed.
Assumes that the node with key is present in the tree.
Examples
1> Tree0 = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> {Value,Tree1} = gb_trees:take(b, Tree0).
3> Value.
2
4> gb_trees:to_list(Tree1).
[{a,1},{c,3}]
-spec take_any(Key, Tree1) -> {Value, Tree2} | error when Tree1 :: tree(Key, _), Tree2 :: tree(Key, _), Key :: term(), Value :: term().
Removes the node with key Key from Tree1 if present; otherwise, returns the tree unchanged.
Examples
1> Tree0 = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> {Value,Tree1} = gb_trees:take_any(b, Tree0).
3> Value.
2
4> gb_trees:to_list(Tree1).
[{a,1},{c,3}]
5> gb_trees:take_any(x, Tree0).
error
-spec take_largest(Tree1) -> {Key, Value, Tree2} when Tree1 :: tree(Key, Value), Tree2 :: tree(Key, Value).
Returns {Key, Value, Tree2}, where Key is the largest key in
Tree1, Value is the value associated with this key, and Tree2 is
this tree with the corresponding node deleted.
Assumes that the tree is not empty.
Examples
1> Tree0 = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> {Key,Value,Tree1} = gb_trees:take_largest(Tree0).
3> Key.
c
4> Value.
3
5> gb_trees:to_list(Tree1).
[{a,1},{b,2}]
-spec take_smallest(Tree1) -> {Key, Value, Tree2} when Tree1 :: tree(Key, Value), Tree2 :: tree(Key, Value).
Returns {Key, Value, Tree2}, where Key is the smallest key in Tree1,
Value is the value associated with that key, and Tree2 is the tree
with the corresponding node removed.
Assumes that the tree is not empty.
Examples
1> Tree0 = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> {Key,Value,Tree1} = gb_trees:take_smallest(Tree0).
3> Key.
a
4> Value.
1
5> gb_trees:to_list(Tree1).
[{b,2},{c,3}]
-spec to_list(Tree) -> [{Key, Value}] when Tree :: tree(Key, Value).
Converts a tree into an ordered list of key-value tuples.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3}]).
2> gb_trees:to_list(Tree).
[{a,1},{b,2},{c,3}]
3> gb_trees:to_list(gb_trees:empty()).
[]
Updates Key to value Value in Tree1 and returns the new tree.
Assumes that the key is present in the tree.
Examples
1> Tree1 = gb_trees:from_orddict([{a,1},{b,2}]).
2> Tree2 = gb_trees:update(a, 99, Tree1).
3> gb_trees:to_list(Tree2).
[{a,99},{b,2}]
Returns the values in Tree as an ordered list, sorted by their
corresponding keys.
Duplicates are not removed.
Examples
1> Tree = gb_trees:from_orddict([{a,1},{b,2},{c,3},{d,1}]).
2> gb_trees:values(Tree).
[1,2,3,1]