java 遍历父子结构的树集合

原创

mob64ca12df277e 2023-10-31 12:09:35 ©著作权

文章标签 子节点树结构 java 文章分类 Java 后端开发

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

Java遍历父子结构的树集合

1. 简介

在开发过程中，经常会遇到需要遍历父子结构的树集合的需求，比如树形菜单、组织机构等。本文将详细介绍如何使用Java来遍历这种树形结构。

2. 流程

下面是整个流程的步骤概述：

flowchart TD
    A(初始化树集合) --> B(遍历树集合)
    B --> C(检查当前节点是否为叶子节点)
    C -- 是 --> D(处理叶子节点)
    C -- 否 --> E(获取子节点列表)
    E --> B

3. 代码实现

首先，我们需要定义一个树节点类，用于表示树结构中的每个节点。节点类包含一个值和一个子节点列表。

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 children;
    }

    // 获取节点值
    public String getValue() {
        return value;
    }

    // 判断是否为叶子节点
    public boolean isLeaf() {
        return children.isEmpty();
    }
}

接下来，我们需要编写一个遍历树集合的方法。这个方法接受一个树节点作为参数，并按照上述流程进行遍历操作。

public void traverseTree(TreeNode root) {
    // 检查当前节点是否为叶子节点
    if (root.isLeaf()) {
        // 处理叶子节点
        System.out.println(root.getValue());
    } else {
        // 获取子节点列表
        List<TreeNode> children = root.getChildren();
        // 遍历子节点列表
        for (TreeNode child : children) {
            // 递归遍历子节点
            traverseTree(child);
        }
    }
}

4. 使用示例

假设我们有如下的树结构：

      A
    /   \
   B     C
  / \   / \
 D   E F   G

我们可以按照以下方式创建树节点并进行遍历：

// 创建根节点
TreeNode root = 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");

// 构建树结构
root.addChild(nodeB);
root.addChild(nodeC);
nodeB.addChild(nodeD);
nodeB.addChild(nodeE);
nodeC.addChild(nodeF);
nodeC.addChild(nodeG);

// 遍历树集合
traverseTree(root);

运行以上代码，将会按照深度优先的方式遍历树结构，并输出每个叶子节点的值：