Java实现公用实体类转Tree结构

戴嘉淑

工具类

1. import java.util.*;
2. import java.util.function.Function;
3. import java.util.stream.Collectors;
5. // 树节点类，用于存储树中的每个元素
6. class TreeNode<T> {
7. private T data; // 节点存储的数据
8. private List<TreeNode<T>> children = new ArrayList<>(); // 子节点列表
10. // 构造函数，初始化节点数据
11. public TreeNode(T data) {
12. this.data = data;
13. }
15. // 添加子节点
16. public void addChild(TreeNode<T> child) {
17. children.add(child);
18. }
20. // 获取当前节点的数据
21. public T getData() {
22. return data;
23. }
25. // 获取当前节点的所有子节点
26. public List<TreeNode<T>> getChildren() {
27. return children;
28. }
29. }
31. // 树构建器类，负责从列表构建树形结构
32. class TreeBuilder<T> {
33. // buildTree方法，构建树形结构
34. public TreeNode<T> buildTree(List<T> items, Function<T, String> idExtractor, Function<T, String> parentIdExtractor) {
35. // 创建一个映射，将每个节点的ID映射到对应的TreeNode对象
36. Map<String, TreeNode<T>> nodeMap = items.stream()
37. .map(item -> new TreeNode<>(item)) // 将每个元素转换为TreeNode
38. .collect(Collectors.toMap(node -> idExtractor.apply(node.getData()), Function.identity()));
40. TreeNode<T> root = null; // 用于存储根节点
42. // 遍历每个元素，建立父子关系
43. for (T item : items) {
44. String id = idExtractor.apply(item); // 获取当前节点的ID
45. String parentId = parentIdExtractor.apply(item); // 获取父节点的ID
46. TreeNode<T> node = nodeMap.get(id); // 获取当前节点
48. if (parentId == null) {
49. root = node; // 假设只有一个根节点
50. } else {
51. TreeNode<T> parentNode = nodeMap.get(parentId); // 获取父节点
52. if (parentNode != null) {
53. parentNode.addChild(node); // 将当前节点添加到父节点的子节点列表中
54. }
55. }
56. }
58. return root; // 返回构建的树的根节点
59. }
60. }

复制代码

注释说明：

TreeNode类：

• 每个字段和方法都有注释，说明其用途。
• 包含构造函数、添加子节点的方法以及获取数据和子节点的方法。

TreeBuilder类：

• buildTree方法的逻辑注释清晰，解释了每一步的目的。
• 使用映射将每个实体转换为树节点，并建立父子关系。

代码示例

1. import java.util.Arrays;
2. import java.util.List;
4. // 实体类，用于表示树节点的数据
5. class Entity {
6. private String id; // 节点ID
7. private String parentId; // 父节点ID
8. private String name; // 节点名称
10. // 构造函数
11. public Entity(String id, String parentId, String name) {
12. this.id = id;
13. this.parentId = parentId;
14. this.name = name;
15. }
17. // 获取节点ID
18. public String getId() {
19. return id;
20. }
22. // 获取父节点ID
23. public String getParentId() {
24. return parentId;
25. }
27. // 获取节点名称
28. public String getName() {
29. return name;
30. }
31. }
33. // 主程序
34. public class Main {
35. public static void main(String[] args) {
36. // 创建实体列表，模拟树形结构
37. List<Entity> entities = Arrays.asList(
38. new Entity("1", null, "Root"), // 根节点
39. new Entity("2", "1", "Child 1"), // Child 1, 父节点为Root
40. new Entity("3", "1", "Child 2"), // Child 2, 父节点为Root
41. new Entity("4", "2", "Grandchild 1") // Grandchild 1, 父节点为Child 1
42. );
44. // 创建树构建器实例
45. TreeBuilder<Entity> treeBuilder = new TreeBuilder<>();
47. // 构建树，传入实体列表及提取ID和父ID的函数
48. TreeNode<Entity> tree = treeBuilder.buildTree(
49. entities,
50. Entity::getId, // 提取ID
51. Entity::getParentId // 提取父ID
52. );
54. // 打印树结构
55. printTree(tree, 0);
56. }
58. // 递归打印树结构的辅助方法
59. private static void printTree(TreeNode<Entity> node, int level) {
60. if (node == null) return; // 如果节点为空，直接返回
62. // 打印当前节点的名称，缩进显示层级
63. System.out.println(" ".repeat(level) + node.getData().getName());
65. // 递归打印所有子节点
66. for (TreeNode<Entity> child : node.getChildren()) {
67. printTree(child, level + 1);
68. }
69. }
70. }

复制代码

代码注释说明：

Entity类：每个字段和方法都有注释，说明其用途和功能。

Main类：

• 实体列表中每个节点的构造都有注释，解释其层级关系。
• 构建树的步骤清晰描述，便于理解。

printTree方法：详细说明其递归逻辑和输出格式。

到此这篇关于Java实现公用实体类转Tree结构的文章就介绍到这了,更多相关Java公用实体类转Tree结构内容请搜索晓枫资讯以前的文章或继续浏览下面的相关文章希望大家以后多多支持晓枫资讯！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

或许他某

感谢楼主分享。

戴帝泰麟

路过，支持一下

瑞安王锋鸦

顶顶更健康！！！