java查询某个节点的所有叶子节点

原创

mob64ca12e98e58 2023-12-15 07:33:10 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e98e58的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java查询某个节点的所有叶子节点

在开发过程中，我们经常需要查询一个节点的所有叶子节点。在Java中，我们可以通过遍历树的方式来实现这个功能。本文将介绍如何使用Java代码查询某个节点的所有叶子节点，并提供相应的代码示例。

关系图

先来看一下我们要处理的节点关系图，如下所示：

erDiagram
    A --|> B
    A --|> C
    B --|> D
    B --|> E
    C --|> F
    C --|> G

在这个关系图中，节点A是根节点，它有两个子节点B和C，B节点有两个子节点D和E，C节点有两个子节点F和G。我们的目标是查询A节点的所有叶子节点。

流程图

下面是查询某个节点的所有叶子节点的流程图：

flowchart TD
    start[开始] --> input[输入节点]
    cond{节点是否为叶子节点？} -->|是| output[输出节点]
    cond -->|否| iterate[遍历子节点]
    iterate --> loop{是否还有子节点？}
    loop --|是| process[处理子节点]
    loop --|否| output[输出节点]
    process --> cond

代码示例

下面是使用Java代码查询某个节点的所有叶子节点的示例：

public class TreeNode {
    private String value;
    private List<TreeNode> children;

    public TreeNode(String value) {
        this.value = value;
        this.children = new ArrayList<>();
    }

    public void addChild(TreeNode child) {
        this.children.add(child);
    }

    public List<TreeNode> getChildren() {
        return this.children;
    }

    public boolean isLeaf() {
        return this.children.isEmpty();
    }

    public List<TreeNode> getAllLeafNodes() {
        List<TreeNode> leafNodes = new ArrayList<>();
        getAllLeafNodesRecursive(this, leafNodes);
        return leafNodes;
    }

    private void getAllLeafNodesRecursive(TreeNode node, List<TreeNode> leafNodes) {
        if (node.isLeaf()) {
            leafNodes.add(node);
        } else {
            for (TreeNode child : node.getChildren()) {
                getAllLeafNodesRecursive(child, leafNodes);
            }
        }
    }
}

public class Main {
    public static void main(String[] args) {
        TreeNode nodeA = new TreeNode("A");
        TreeNode nodeB = new TreeNode("B");
        TreeNode nodeC = new TreeNode("C");
        TreeNode nodeD = new TreeNode("D");
        TreeNode nodeE = new TreeNode("E");
        TreeNode nodeF = new TreeNode("F");
        TreeNode nodeG = new TreeNode("G");

        nodeA.addChild(nodeB);
        nodeA.addChild(nodeC);
        nodeB.addChild(nodeD);
        nodeB.addChild(nodeE);
        nodeC.addChild(nodeF);
        nodeC.addChild(nodeG);

        List<TreeNode> leafNodes = nodeA.getAllLeafNodes();

        for (TreeNode leafNode : leafNodes) {
            System.out.println(leafNode.getValue());
        }
    }
}

在上面的代码示例中，我们定义了一个TreeNode类，表示树的节点。每个节点包含一个值和它的子节点列表。我们使用递归的方式来查询某个节点的所有叶子节点。在getAllLeafNodesRecursive方法中，如果当前节点是叶子节点，则将其添加到结果列表中；否则，递归地处理每个子节点。

在Main类的main方法中，我们创建了一个节点关系图，并查询根节点A的所有叶子节点。然后，我们遍历并输出这些叶子节点的值。