java - 如何在二叉树中搜索一个节点并返回它？

Question

我正在尝试在二叉树中搜索一个节点，如果存在则返回，否则返回 null。顺便说一句，节点类有一个方法 name() 返回一个带有它的名字的字符串......我到目前为止是:

private Node search(String name, Node node){

     if(node != null){
         if(node.name().equals(name)){
            return node;
         }

      else{
         search(name, node.left);
         search(name, node.right);
      }
    }
    return null;
}

这是正确的吗？？

Answer 1

如果结果不为空，您需要确保对搜索的递归调用返回。

像这样的东西应该有用...

private Node search(String name, Node node){
    if(node != null){
        if(node.name().equals(name)){
           return node;
        } else {
            Node foundNode = search(name, node.left);
            if(foundNode == null) {
                foundNode = search(name, node.right);
            }
            return foundNode;
         }
    } else {
        return null;
    }
}