在二叉查找树中的查找,类似于在数组上的二分查找,但二叉查找树是查找两棵子树中的一棵,而不是查找数组的一半。
虽然用树和子树很容易表示递归算法,但根节点同样可以用来查找或操作其后代结点。
下面是算法,它的描述接近于实际的实现:
Algorithm bstSearch(binarySearchTreeRoot, desiredObject)
// 在一颗二叉查找树中查找给定的对象
// 如果找到对象则返回真
if(binarySearchTreeRoot是空的)
return false
else if (desiredObject == binarySearchTreeRoot中的对象)
return true
else if (desiredObject < binarySearchTreeRoot中的对象)
return bstSearch(binarySearchTreeRoot的左孩子, desiredObject)
else
return bstSearch(binarySearchTreeRoot的右孩子, desiredObject)
类似于递归算法,我们将实际的查找过程实现为一个私有方法findEntry,而由公有方法getEntry来调用。虽然该算法返回一个布尔值,但实现返回确定了数据对象。方法如下:
public T getEntry(T entry)
{
return findEntry(getRootNode(), entry);
}
private T findEntry(BinaryNode<T> rootNode, T entry)
{
T result = null;
if(rootNode != null)
{
T rootEntry = rootNode.getData();
if (entry.equals(rootEntry))
result = rootEntry;
else if (entry.compareTo(rootEntry) < 0)
result = findEntry(rootNode.getLeftChild(), entry);
else
result = findEntry(rootNode.getRightChild(), entry);
}
return result;
}
方法contains可简单地调用getEntry,看看给定的项是否在树中:
public boolean contains(T entry)
{
return getEntry(entry) != null;
}