这是一个家庭作业。我需要递归遍历一个二叉搜索树,找出一个节点的数据值是否在一个范围内(包括),并按升序打印出来。
我的思考过程是:要按升序打印数据值,我需要在搜索树上按顺序遍历。为了有效地遍历,如果一个节点的左孩子小于下限值,并且该节点的数据小于下限值,程序应该停止向左遍历。同理,如果一个节点的右孩子大于上限值,并且该节点的数据大于上限值,程序应该停止向右遍历。
所以这是我的实现,但有错误:
public void rangeSearch(int lower, int upper) {
if (lower > upper)
throw new IllegalArgumentException("lower > upper");
if (root != null)
rangeSearchTree(root, lower, upper);
}
private static void rangeSearchTree(Node root, int lower, int upper) {
Node leftChild = root.left;
Node rightChild = root.right;
if (leftChild != null && root.key > lower) {
root = leftChild;
rangeSearchTree(root, lower, upper);
}
if (root.key >= lower && root.key <= upper) {
System.out.print(root.key + " ");
}
if (rightChild != null && root.key < upper) {
root = rightChild;
rangeSearchTree(root, lower, upper);
}
}
树的结构如下:
7
/ \
5 9
/ \ /
2 6 8
\
4
当我输入6
下限值和9
上限值时,我得到6 8 8
. 正确答案应该是6 7 8 9
。有什么建议么?