java-BST递归查找值

Question

我有一棵 BST 树。我想创建一个方法来获取一个值并返回包含它的值的节点的级别（root = 0），没有这样的节点吗？返回-1。 我想递归地做。 这段代码工作得很好：

    private int recursiveContains(BinaryNode node, int searchVal){
    int nodeKey = node.nodeKey;
    if (searchVal < nodeKey){
        if (node.leftChild != EMPTY_NODE)
            return 1 + recursiveContains(node.leftChild, searchVal);
    }else if (searchVal > nodeKey){
        if (node.rightChild != EMPTY_NODE)
            return 1 + recursiveContains(node.rightChild, searchVal);
    }
    return 0;
}

但是，只要树包含搜索值。

当我到达叶子但没有找到值时，如何停止迭代并返回 -1？ 可以递归吗？

谢谢

Answer 1

您只需要调整您的最终案例。现在，如果该值不在树中，您只需返回将插入该值的节点的深度，因为您的最终情况只是return 0。相反，您需要明确检查当前节点是否确实是正确的节点。如果是，可以返回0；否则你应该返回-1。然后，递归调用需要查找该特殊值并适当地处理它。

我可能会把这个明确的检查——这是请求节点的基本情况——放在开头。最后，您的“失败”值（如果没有其他条件为真，则返回）是-1。所以你最终会得到这样的结果：

// WARNING: UNTESTED CODE
if (searchVal == nodeKey) {
    return 0;
} else if (searchVal < nodeKey && node.leftChild != EMPTY_NODE) {
    int childResult = recursiveContains(node.leftChild, searchVal);
    if (childResult != -1) { // Only use the child result if the value was found.
        return 1 + childResult;
    }
} else if (searchVal > nodeKey && node.rightChild != EMPTY_NODE) {
    int childResult = recursiveContains(node.rightChild, searchVal);
    if (childResult != -1) { // Only use the child result if the value was found.
        return 1 + childResult;
    }
}
// If you haven't returned by now, the value can't be found along this path.
return -1;