Java:找到BST中未找到的节点的深度

时间:2014-10-20 15:20:44

标签: java binary-search-tree depth

我有一个存储在字符串中的二叉搜索树,我已经能够找到BST中存在的字符串的深度或高度,但我想要做的是找到深度将是什么不在BST中的字符串。

这是我目前正在使用的算法,它可以很好地找到BST中已经存在的字符串的深度:

public int getDepth(String data) {
    return getDepth(root, data, 1);
}

public int getDepth(Node root, String data, int level) {
    if (root == null)
        return 0;

    if (root.getData().getName().equals(data))
        return level;

    int ret = getDepth(root.leftChild, data, level + 1);
    if (ret != 0)
        return ret;

    ret = getDepth(root.rightChild, data, level + 1);
    return ret;

}

我的BST有按以下顺序插入的元素:“cat”,“bird”,“dog”,“tiger”,“elephant”,“panda”。

所以二叉树看起来应该是这样的:

                                   cat
                                  /   \
                               bird   dog
                                         \
                                        tiger
                                       /     \
                                  elephant  panda

当我在“dog”上调用方法时,输出为:dog深度为2

当我在“hippo”上调用该方法时,输出为:hippo在深度0

但“河马”的预期输出应该是:河马深度为5

因为河马将是已经深入4的大象的孩子。

那么对于另一个案例,如果我想找到不在BST中的“牛”的深度怎么办?它必须是狗的左孩子,所以深度应该是3,但是,我仍然得到0。

所以问题是没有使用删除和/或恢复方法我需要什么情况或我需要在上面的方法中修改哪些情况以确定不在里面的字符串的深度BST?

2 个答案:

答案 0 :(得分:2)

你需要模拟insert操作,该操作执行真实插入所做的所有事情,除非它不向树附加新节点并跟踪深度。
搜索正确的位置以插入新值 - 从根开始,将值与当前节点进行比较,根据比较结果下降树。 找到正确的位置后,不要将值添加到树中,只需返回深度。

答案 1 :(得分:1)

尝试使用第二个getDepth()方法的修改版本:

public int getDepth(Node root, String data) {
    if (root == null || root.getData().getName().equals(data)) {
        // you hit the spot - either you are below a leaf
        // or you matched the node
        // if you were doing insert/replace, you would put the data on this spot
        return 0;
    }

    if (isSmallerThan(data, root.getData().getName())) {
        return 1 + getDepth(root.leftChild, data);
    }
    return 1 + getDepth(root.rightChild, data);
}

如果isSmallerThan(String a, String b) 小于而不是true,则a会返回b。此方法取决于您希望如何比较字符串。

而且你不需要一直传递level ......