rbtrees.pl -- Red black trees
Red-Black trees are balanced search binary trees. They are named because nodes can be classified as either red or black. The code we include is based on "Introduction to Algorithms", second edition, by Cormen, Leiserson, Rivest and Stein. The library includes routines to insert, lookup and delete elements in the tree.
A Red black tree is represented as a term t(Nil, Tree)
, where Nil is the
Nil-node, a node shared for each nil-node in the tree. Any node has the
form colour(Left, Key, Value, Right)
, where colour is one of red
or
black
.
Warning: instantiation of keys
Red-Black trees depend on the Prolog standard order of terms to organize the keys as a (balanced) binary tree. This implies that any term may be used as a key. The tree may produce wrong results, such as not being able to find a key, if the ordering of keys changes after the key has been inserted into the tree. The user is responsible to ensure that variables used as keys or appearing in a term used as key that may affect ordering are not unified, with the exception of unification against new fresh variables. For this reason, ground terms are safe keys. When using non-ground terms, either make sure the variables appear in places that do not affect the standard order relative to other keys in the tree or make sure to not unify against these variables as long as the tree is being used.
- rb_new(-Tree) is det
- Create a new Red-Black tree Tree.
- rb_empty(?Tree) is semidet
- Succeeds if Tree is an empty Red-Black tree.
- rb_lookup(+Key, -Value, +Tree) is semidet
- True when Value is associated with Key in the Red-Black tree Tree. The given Key may include variables, in which case the RB tree is searched for a key with equivalent variables (using (==)/2). Time complexity is O(log N) in the number of elements in the tree.
- rb_min(+Tree, -Key, -Value) is semidet
- Key is the minimum key in Tree, and is associated with Val.
- rb_max(+Tree, -Key, -Value) is semidet
- Key is the maximal key in Tree, and is associated with Val.
- rb_next(+Tree, +Key, -Next, -Value) is semidet
- Next is the next element after Key in Tree, and is associated with Val. Fails if Key isn't in Tree or if Key is the maximum key.
- rb_previous(+Tree, +Key, -Previous, -Value) is semidet
- Previous is the previous element after Key in Tree, and is associated with Val. Fails if Key isn't in Tree or if Key is the minimum key.
- rb_update(+Tree, +Key, ?NewVal, -NewTree) is semidet
- Tree NewTree is tree Tree, but with value for Key associated with NewVal. Fails if Key is not in Tree (using (==)/2). This predicate may fail or give unexpected results if Key is not sufficiently instantiated.
- rb_update(+Tree, +Key, -OldVal, ?NewVal, -NewTree) is semidet
- Same as
rb_update(Tree, Key, NewVal, NewTree)
but also unifies OldVal with the value associated with Key in Tree. - rb_apply(+Tree, +Key, :G, -NewTree) is semidet
- If the value associated with key Key is Val0 in Tree, and if
call(G,Val0,ValF)
holds, then NewTree differs from Tree only in that Key is associated with value ValF in tree NewTree. Fails if it cannot find Key in Tree, or ifcall(G,Val0,ValF)
is not satisfiable. - rb_in(?Key, ?Value, +Tree) is nondet
- True when Key-Value is a key-value pair in red-black tree Tree. Same
as below, but does not materialize the pairs.
rb_visit(Tree, Pairs), member(Key-Value, Pairs)
Leaves a choicepoint even if Key is instantiated; to avoid a choicepoint, use rb_lookup/3.
- rb_insert(+Tree, +Key, ?Value, -NewTree) is det
- Add an element with key Key and Value to the tree Tree creating a new red-black tree NewTree. If Key is a key in Tree, the associated value is replaced by Value. See also rb_insert_new/4. Does not validate that Key is sufficiently instantiated to ensure the tree remains valid if a key is further instantiated.
- rb_insert_new(+Tree, +Key, ?Value, -NewTree) is semidet
- Add a new element with key Key and Value to the tree Tree creating a new red-black tree NewTree. Fails if Key is a key in Tree. Does not validate that Key is sufficiently instantiated to ensure the tree remains valid if a key is further instantiated.
- rb_delete(+Tree, +Key, -NewTree)
- Delete element with key Key from the tree Tree, returning the value Val associated with the key and a new tree NewTree. Fails if Key is not in Tree (using (==)/2).
- rb_delete(+Tree, +Key, -Val, -NewTree)
- Same as
rb_delete(Tree, Key, NewTree)
, but also unifies Val with the value associated with Key in Tree. - rb_del_min(+Tree, -Key, -Val, -NewTree)
- Delete the least element from the tree Tree, returning the key Key, the value Val associated with the key and a new tree NewTree. Fails if Tree is empty.
- rb_del_max(+Tree, -Key, -Val, -NewTree)
- Delete the largest element from the tree Tree, returning the key Key, the value Val associated with the key and a new tree NewTree. Fails if Tree is empty.
- rb_visit(+Tree, -Pairs) is det
- Pairs is an infix visit of tree Tree, where each element of Pairs is of the form Key-Value.
- rb_map(+T, :Goal) is semidet
- True if
call(Goal, Value)
is true for all nodes in T. - rb_map(+Tree, :G, -NewTree) is semidet
- For all nodes Key in the tree Tree, if the value associated with key
Key is Val0 in tree Tree, and if
call(G,Val0,ValF)
holds, then the value associated with Key in NewTree is ValF. Fails ifcall(G,Val0,ValF)
is not satisfiable for all Val0. If G is non-deterministic, rb_map/3 will backtrack over all possible values fromcall(G,Val0,ValF)
. You should not depend on the order of tree traversal (currently: key order). - rb_fold(:Goal, +Tree, +State0, -State)
- Fold the given predicate over all the key-value pairs in Tree,
starting with initial state State0 and returning the final state
State. Pred is called as
call(Pred, Key-Value, State1, State2)
Determinism depends on Goal.
- rb_clone(+TreeIn, -TreeOut, -Pairs) is det
- `Clone' the red-back tree TreeIn into a new tree TreeOut with the same keys as the original but with all values set to unbound values. Pairs is a list containing all new nodes as pairs K-V.
- rb_partial_map(+Tree, +Keys, :G, -NewTree)
- For all nodes Key in Keys, if the value associated with key Key is
Val0 in tree Tree, and if
call(G,Val0,ValF)
holds, then the value associated with Key in NewTree is ValF, otherwise it is the value associated with the key in Tree. Fails if Key isn't in Tree or ifcall(G,Val0,ValF)
is not satisfiable for all Val0 in Keys. Assumes keys are sorted and not repeated (fails if this is not true). - rb_keys(+Tree, -Keys) is det
- Keys is unified with an ordered list of all keys in the Red-Black tree Tree.
- list_to_rbtree(+List, -Tree) is det
- Tree is the red-black tree corresponding to the mapping in List, which should be a list of Key-Value pairs. List should not contain more than one entry for each distinct key, but this is not validated by list_to_rbtree/2.
- ord_list_to_rbtree(+List, -Tree) is det
- Tree is the red-black tree corresponding to the mapping in list List, which should be a list of Key-Value pairs. List should not contain more than one entry for each distinct key, but this is not validated by ord_list_to_rbtree/2. List is assumed to be sorted according to the standard order of terms.
- rb_size(+Tree, -Size) is det
- Size is the number of elements in Tree.
- is_rbtree(@Term) is semidet
- True if Term is a valid Red-Black tree. Processes the entire tree, checking the coloring of the nodes, the balance and the ordering of keys. Does not validate that keys are sufficiently instantiated to ensure the tree remains valid if a key is further instantiated.