如何在树中搜索节点并将其返回?

bes*_*ico 11 java binary-tree tree-nodes

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

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;
}
Run Code Online (Sandbox Code Playgroud)

它是否正确??

Tom*_*rys 25

如果结果不为null,则需要确保对搜索的递归调用返回.

像这样的东西应该工作......

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;
    }
}
Run Code Online (Sandbox Code Playgroud)