java list转树形结构

原创

mob64ca12f31496 2023-09-05 18:16:01 ©著作权

文章标签 List 树形结构 Java 文章分类 Java 后端开发

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

Java List转树形结构的实现

1. 简介

在Java开发中，经常会遇到将一个List集合转换成树形结构的需求。例如，将数据库中的多级分类数据进行展示时，就需要将数据库查询出的平级数据转换成树形结构进行展示。

本文将介绍如何实现Java List转树形结构的功能，包括具体的步骤和所需的代码。

2. 实现步骤

下面是实现Java List转树形结构的步骤：

步骤	描述
步骤1	创建树节点类
步骤2	构建树节点之间的父子关系
步骤3	将List集合转换成树形结构

下面将详细介绍每个步骤的具体实现方法。

3. 具体实现

步骤1：创建树节点类

首先，我们需要创建一个树节点类，用于表示树中的每个节点。该类包含一个值和一个子节点列表。

public class TreeNode<T> {
    private T value; // 节点值
    private List<TreeNode<T>> children; // 子节点列表

    // 构造方法
    public TreeNode(T value) {
        this.value = value;
        this.children = new ArrayList<>();
    }

    // Getter和Setter方法
    // ...
}

步骤2：构建树节点之间的父子关系

接下来，我们需要根据父子关系来构建树节点之间的连接。假设我们的List集合中的每个元素都包含一个父节点ID和一个当前节点ID。我们可以使用一个Map来存储节点ID和对应的节点对象，然后遍历List集合，将每个节点对象添加到对应的父节点的子节点列表中。

// 假设List集合中的每个元素都包含parentID和nodeID两个字段

List<TreeNode<T>> nodeList = new ArrayList<>(); // 存储节点对象的列表
Map<T, TreeNode<T>> nodeMap = new HashMap<>(); // 存储节点ID和节点对象的映射关系

// 遍历List集合，创建节点对象并添加到nodeList和nodeMap中
for (NodeData<T> nodeData : dataList) {
    TreeNode<T> node = new TreeNode<>(nodeData.nodeID);
    nodeList.add(node);
    nodeMap.put(nodeData.nodeID, node);
}

// 遍历List集合，构建父子关系
for (NodeData<T> nodeData : dataList) {
    TreeNode<T> node = nodeMap.get(nodeData.nodeID);
    TreeNode<T> parent = nodeMap.get(nodeData.parentID);
    parent.getChildren().add(node);
}

步骤3：将List集合转换成树形结构

最后，我们需要将节点列表转换成树形结构。由于树形结构是有且仅有一个根节点的，我们可以通过遍历节点列表，找到没有父节点的节点作为根节点，然后从根节点开始递归地构建树。

TreeNode<T> root = null; // 根节点

// 找到没有父节点的节点作为根节点
for (TreeNode<T> node : nodeList) {
    if (node.getParent() == null) {
        root = node;
        break;
    }
}

// 递归地构建树
buildTree(root);

// 递归函数
private void buildTree(TreeNode<T> node) {
    for (TreeNode<T> child : node.getChildren()) {
        buildTree(child);
    }
}

以上就是实现Java List转树形结构的全部步骤和代码。