Java利用TreeUtils工具类实现列表转树

下面是Java利用TreeUtils工具类实现列表转树的完整攻略。

1.准备工作

在进行列表转树操作前，需要先准备好列表数据。假设列表中每个元素都具有一个唯一标识符id和一个父元素标识符parentId，我们可以封装一个类来表示列表元素：

public class TreeNode {
    private String id;
    private String parentId;
    // 其他属性

    // 构造方法、getter、setter等略
}

同时为了方便读取列表数据，我们可以将列表数据读取到一个List集合中。为了使用TreeUtils工具类，我们还需要添加相关依赖：

<dependency>
    <groupId>com.github.hiwepy</groupId>
    <artifactId>commons-treetable-utils</artifactId>
    <version>1.0.0</version>
</dependency>

2.列表转树

接下来，就可以利用TreeUtils工具类将列表转换成对应的树形结构。TreeUtils提供了常见的几种转换方式，例如：层级关系转树结构、扁平化树结构转嵌套树结构等。这里我们以层级关系转树结构为例进行讲解。

List<TreeNode> nodeList = new ArrayList<>(); // 待转换的列表数据
Map<String, TreeNode> nodeMap = new HashMap<>();
for (TreeNode node : nodeList) {
    nodeMap.put(node.getId(), node);
}

TreeModel<TreeNode> model = new DefaultTreeModel<>(); // 新建一个空树
for (TreeNode node : nodeMap.values()) {
    TreeNode parentNode = nodeMap.get(node.getParentId());
    if (parentNode != null) {
        model.addLeaf(parentNode, node); // 将子节点添加到父节点下
    } else {
        model.setRoot(node); // 将节点作为根节点
    }
}

List<TreeNode> treeList = new ArrayList<>();
List<TreeNode> rootList = model.getRootList();
for (TreeNode root : rootList) {
    treeList.add(root);
    treeList.addAll(model.getChildrenList(root)); // 将子节点添加到树列表中
}

以上代码将把nodeList列表转换成树形结构，得到的树形结构以List的形式返回，其中节点顺序为先父后子。下面是一个示例：

假设nodeList中有四个元素，分别是 {"1","-1"},{"2","1"},{"3","1"},{"4","2"}，其中第一个元素的parentId为-1，表示它是根节点。

使用上述代码后，最终的树形结构如下：

1
├──2
│  └──4
└──3

其中，每个节点的id和parentId已在TreeNode类中定义。

3.示例

下面我们举两个示例来说明使用TreeUtils工具类实现列表转树的过程。

示例1

假设列表数据为：

TreeNode node1 = new TreeNode("1", "-1");
TreeNode node2 = new TreeNode("2", "1");
TreeNode node3 = new TreeNode("3", "1");
TreeNode node4 = new TreeNode("4", "2");

使用TreeUtils工具类将其转换成树形结构的代码如下：

List<TreeNode> nodeList = Arrays.asList(node1, node2, node3, node4);
Map<String, TreeNode> nodeMap = new HashMap<>();
for (TreeNode node : nodeList) {
    nodeMap.put(node.getId(), node);
}

TreeModel<TreeNode> model = new DefaultTreeModel<>();
for (TreeNode node : nodeMap.values()) {
    TreeNode parentNode = nodeMap.get(node.getParentId());
    if (parentNode != null) {
        model.addLeaf(parentNode, node);
    } else {
        model.setRoot(node);
    }
}

List<TreeNode> treeList = new ArrayList<>();
List<TreeNode> rootList = model.getRootList();
for (TreeNode root : rootList) {
    treeList.add(root);
    treeList.addAll(model.getChildrenList(root));
}

System.out.println(treeList);

输出结果为：

[TreeNode(id=1, parentId=-1), TreeNode(id=2, parentId=1), TreeNode(id=4, parentId=2), TreeNode(id=3, parentId=1)]

说明转换成功，树形结构如下：

1
├──2
│  └──4
└──3

示例2

假设列表数据为：

TreeNode node1 = new TreeNode("1", "-1");
TreeNode node2 = new TreeNode("2", "1");
TreeNode node3 = new TreeNode("3", "2");
TreeNode node4 = new TreeNode("4", "2");

使用TreeUtils工具类将其转换成树形结构的代码如下：

List<TreeNode> nodeList = Arrays.asList(node1, node2, node3, node4);
Map<String, TreeNode> nodeMap = new HashMap<>();
for (TreeNode node : nodeList) {
    nodeMap.put(node.getId(), node);
}

TreeModel<TreeNode> model = new DefaultTreeModel<>();
for (TreeNode node : nodeMap.values()) {
    TreeNode parentNode = nodeMap.get(node.getParentId());
    if (parentNode != null) {
        model.addLeaf(parentNode, node);
    } else {
        model.setRoot(node);
    }
}

List<TreeNode> treeList = new ArrayList<>();
List<TreeNode> rootList = model.getRootList();
for (TreeNode root : rootList) {
    treeList.add(root);
    treeList.addAll(model.getChildrenList(root));
}

System.out.println(treeList);

输出结果为：

[TreeNode(id=1, parentId=-1), TreeNode(id=2, parentId=1), TreeNode(id=3, parentId=2), TreeNode(id=4, parentId=2)]

说明转换成功，树形结构如下：

1
└──2
   ├──3
   └──4

这两个示例演示了如何使用TreeUtils工具类实现列表转树的过程，具体方法和顺序请参照以下步骤：

1. 准备列表数据，将其封装成节点对象；
2. 新建一个空树；
3. 遍历列表节点，将子节点添加到父节点下，或者将节点作为根节点；
4. 将转换后的树形结构封装成List返回。

希望这个攻略能够帮助你。