如何在非二叉树中找到特定节点？

Question

我有一个包含多个节点的树。 每个节点都有一个父节点（在根节点的情况下为 null）、一个名称（名称）和一个将子节点的名称映射到子节点对象的哈希表（子节点）。

给定节点的字符串名称，我想创建一个方法来遍历树以查找特定节点并返回它。 如果节点不存在，则返回 null。

我认为递归方法是最好的。 到目前为止，我有这个：

public Node findNode(Node n, String s) {
   if (n.name == s) {
       return n;
   } else {
       for (Node child: n.children.values()) {
            findNode(child, s);
       }
     }
}

我不确定将 null 语句放在哪里。

Answer 1

如果孩子有它，然后将其归还。 如果没有，则返回null 。

public Node findNode(Node n, String s) {
    if (n.name == s) {
        return n;
    } else {
        for (Node child: n.children.values()) {
            Node result = findNode(child, s);
            if (result != null) {
                return result;
            }
        }
    }
    return null;
}

Answer 2

这是使用Optional搜索节点的 Java 11+ 版本。

public Optional<Node> search (Node node, String needle) {
    if (node.getValue().equals(needle)) {
        return Optional.of(node);
    } else {
        for (var child : node.getChildren()) {
            var result = search(child, needle);
            if (result.isPresent()) {
                return result;
            }
        }
    }

    return Optional.empty();
}