# BK-tree

A BK-tree is a metric tree suggested by Walter Austin Burkhard and Robert M. Keller specifically adapted to discrete metric spaces. For simplicity, consider integer discrete metric $\displaystyle{ d(x,y) }$. Then, BK-tree is defined in the following way. An arbitrary element a is selected as root node. The root node may have zero or more subtrees. The k-th subtree is recursively built of all elements b such that $\displaystyle{ d(a,b) = k }$. BK-trees can be used for approximate string matching in a dictionary.

## Example

This picture depicts the BK-tree for the set $\displaystyle{ W }$ of words {"book", "books", "cake", "boo", "boon", "cook", "cake", "cape", "cart"} obtained by using the Levenshtein distance

• each node $\displaystyle{ u }$ is labeled by a string of $\displaystyle{ w_u \in W }$;
• each arc $\displaystyle{ (u,v) }$ is labeled by $\displaystyle{ d_{uv} = d(w_u,w_v) }$ where $\displaystyle{ w_u }$ denotes the word assigned to $\displaystyle{ u }$.

The BK-tree is built so that:

• for all node $\displaystyle{ u }$ of the BK-tree, the weight assigned to its egress arcs are distinct;
• for all arc $\displaystyle{ e=(u,v) }$ labeled by $\displaystyle{ k }$, each descendant $\displaystyle{ v' }$ of $\displaystyle{ v }$ satisfies the following equation: $\displaystyle{ d(w_u, w_{v'}) = k }$:
• Example 1: Consider the arc from "book" to "books". The distance between "book" and any word in {"books", "boo", "boon", "cook"} is equal to 1;
• Example 2: Consider the arc from "books" to "boo". The distance between "books" and any word in {"boo", "boon", "cook"} is equal to 2.

## Insertion

The insertion primitive is used to populate a BK-tree $\displaystyle{ t }$ according to a discrete metric $\displaystyle{ d }$.

Input:

• $\displaystyle{ t }$: the BK-tree;
• $\displaystyle{ d_{uv} }$ denotes the weight assigned to an arc $\displaystyle{ (u, v) }$;
• $\displaystyle{ w_u }$ denotes word assigned to a node $\displaystyle{ u }$);
• $\displaystyle{ d }$: the discrete metric used by $\displaystyle{ t }$ (e.g. the Levenshtein distance);
• $\displaystyle{ w }$: the element to be inserted into $\displaystyle{ t }$;

Output:

• The node of $\displaystyle{ t }$ corresponding to $\displaystyle{ w }$

Algorithm:

• If the $\displaystyle{ t }$ is empty:
• Create a root node $\displaystyle{ r }$ in $\displaystyle{ t }$
• $\displaystyle{ w_r \leftarrow w }$
• Return $\displaystyle{ r }$
• Set $\displaystyle{ u }$ to the root of $\displaystyle{ t }$
• While $\displaystyle{ u }$ exists:
• $\displaystyle{ k \leftarrow d(w_u, w) }$
• If $\displaystyle{ k = 0 }$:
• Return $\displaystyle{ u }$
• Find $\displaystyle{ v }$ the child of $\displaystyle{ u }$ such that $\displaystyle{ d_{uv} = k }$
• If $\displaystyle{ v }$ is not found:
• Create the node $\displaystyle{ v }$
• $\displaystyle{ w_v \leftarrow w }$
• Create the arc $\displaystyle{ (u, v) }$
• $\displaystyle{ d_{uv} \leftarrow k }$
• Return $\displaystyle{ v }$
• $\displaystyle{ u \leftarrow v }$

## Lookup

Given a searched element $\displaystyle{ w }$, the lookup primitive traverses the BK-tree to find the closest element of $\displaystyle{ w }$. The key idea is to restrict the exploration of $\displaystyle{ t }$ to nodes that can only improve the best candidate found so far by taking advantage of the BK-tree organization and of the triangle inequality (cut-off criterion).

Input:

• $\displaystyle{ t }$: the BK-tree;
• $\displaystyle{ d }$: the corresponding discrete metric (e.g. the Levenshtein distance);
• $\displaystyle{ w }$: the searched element;
• $\displaystyle{ d_{max} }$: the maximum distance allowed between the best match and $\displaystyle{ w }$, defaults to $\displaystyle{ +\infty }$;

Output:

• $\displaystyle{ w_{best} }$: the closest element to $\displaystyle{ w }$ stored in $\displaystyle{ t }$ and according to $\displaystyle{ d }$ or $\displaystyle{ \perp }$ if not found;

Algorithm:

• If $\displaystyle{ t }$ is empty:
• Return $\displaystyle{ \perp }$
• Create $\displaystyle{ S }$ a set of nodes to process, and insert the root of $\displaystyle{ t }$ into $\displaystyle{ S }$.
• $\displaystyle{ (w_{best}, d_{best}) \leftarrow (\perp, d_{max}) }$
• While $\displaystyle{ S \ne \emptyset }$:
• Pop an arbitrary node $\displaystyle{ u }$ from $\displaystyle{ S }$
• $\displaystyle{ d_u \leftarrow d(w, w_u) }$
• If $\displaystyle{ d_u \lt d_{best} }$:
• $\displaystyle{ (w_{best}, d_{best}) \leftarrow (w_u, d_u) }$
• For each egress-arc $\displaystyle{ (u, v) }$:
• If $\displaystyle{ |d_{uv} - d_u| \lt d_{best} }$: (cut-off criterion)
• Insert $\displaystyle{ v }$ into $\displaystyle{ S }$.
• Return $\displaystyle{ w_{best} }$

## Example of the Lookup Algorithm

Consider the example 8-node B-K Tree shown above and set $\displaystyle{ w= }$"cool". $\displaystyle{ S }$ is initialized to contain the root of the tree, which is subsequently popped as the first value of $\displaystyle{ u }$ with $\displaystyle{ w_u }$="book". Further $\displaystyle{ d_u=2 }$ since the distance from "book" to "cool" is 2, and $\displaystyle{ d_{best}=2 }$ as this is the best (i.e. smallest) distance found thus far. Next each outgoing arc from the root is considered in turn: the arc from "book" to "books" has weight 1, and since $\displaystyle{ |1-2|=1 }$ is less than $\displaystyle{ d_{best}=2 }$, the node containing "books" is inserted into $\displaystyle{ S }$ for further processing. The next arc, from "book" to "cake," has weight 4, and since $\displaystyle{ |4-2|=2 }$ is not less than $\displaystyle{ d_{best}=2 }$, the node containing "cake" is not inserted into $\displaystyle{ S }$. Therefore, the subtree rooted at "cake" will be pruned from the search, as the word closest to "cool" cannot appear in that subtree. To see why this pruning is correct, notice that a candidate word $\displaystyle{ c }$ appearing in "cake"s subtree having distance less than 2 to "cool" would violate the triangle inequality: the triangle inequality requires that for this set of three numbers (as sides of a triangle), no two can sum to less than the third, but here the distance from "cool" to "book" (which is 2) plus the distance from "cool" to $\displaystyle{ c }$ (which is less than 2) cannot reach or exceed the distance from "book" to "cake" (which is 4). Therefore, it is safe to disregard the entire subtree rooted at "cake".

Next the node containing "books" is popped from $\displaystyle{ S }$ and now $\displaystyle{ d_u=3 }$, the distance from "cool" to "books." As $\displaystyle{ d_u \gt d_{best} }$, $\displaystyle{ d_{best} }$ remains set at 2 and the single outgoing arc from the node containing "books" is considered. Next, the node containing "boo" is popped from $\displaystyle{ S }$ and $\displaystyle{ d_u=2 }$, the distance from "cool" to "boo." This again does not improve upon $\displaystyle{ d_{best} = 2 }$. Each outgoing arc from "boo" is now considered; the arc from "boo" to "boon" has weight 1, and since $\displaystyle{ |2-1|=1 \lt d_{best}=2 }$, "boon" is added to $\displaystyle{ S }$. Similarly, since $\displaystyle{ |2-2|=0 \lt d_{best} }$, "cook" is also added to $\displaystyle{ S }$.

Finally each of the two last elements in $\displaystyle{ S }$ are considered in arbitrary order: suppose the node containing "cook" is popped first, improving $\displaystyle{ d_{best} }$ to distance 1, then the node containing "boon" is popped last, which has distance 2 from "cool" and therefore does not improve the best result. Finally, "cook" is returned as the answer $\displaystyle{ w_{best} }$ with $\displaystyle{ d_{best}=1 }$.