工具类
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;// 树节点类,用于存储树中的每个元素
class TreeNode<T> {private T data; // 节点存储的数据private List<TreeNode<T>> children = new ArrayList<>(); // 子节点列表// 构造函数,初始化节点数据public TreeNode(T data) {this.data = data;}// 添加子节点public void addChild(TreeNode<T> child) {children.add(child);}// 获取当前节点的数据public T getData() {return data;}// 获取当前节点的所有子节点public List<TreeNode<T>> getChildren() {return children;}
}// 树构建器类,负责从列表构建树形结构
class TreeBuilder<T> {// buildTree方法,构建树形结构public TreeNode<T> buildTree(List<T> items, Function<T, String> idExtractor, Function<T, String> parentIdExtractor) {// 创建一个映射,将每个节点的ID映射到对应的TreeNode对象Map<String, TreeNode<T>> nodeMap = items.stream().map(item -> new TreeNode<>(item)) // 将每个元素转换为TreeNode.collect(Collectors.toMap(node -> idExtractor.apply(node.getData()), Function.identity()));TreeNode<T> root = null; // 用于存储根节点// 遍历每个元素,建立父子关系for (T item : items) {String id = idExtractor.apply(item); // 获取当前节点的IDString parentId = parentIdExtractor.apply(item); // 获取父节点的IDTreeNode<T> node = nodeMap.get(id); // 获取当前节点if (parentId == null) {root = node; // 假设只有一个根节点} else {TreeNode<T> parentNode = nodeMap.get(parentId); // 获取父节点if (parentNode != null) {parentNode.addChild(node); // 将当前节点添加到父节点的子节点列表中}}}return root; // 返回构建的树的根节点}
}
注释说明:
-
TreeNode类:
- 每个字段和方法都有注释,说明其用途。
- 包含构造函数、添加子节点的方法以及获取数据和子节点的方法。
-
TreeBuilder类:
buildTree
方法的逻辑注释清晰,解释了每一步的目的。- 使用映射将每个实体转换为树节点,并建立父子关系。
代码示例
import java.util.Arrays;
import java.util.List;// 实体类,用于表示树节点的数据
class Entity {private String id; // 节点IDprivate String parentId; // 父节点IDprivate String name; // 节点名称// 构造函数public Entity(String id, String parentId, String name) {this.id = id;this.parentId = parentId;this.name = name;}// 获取节点IDpublic String getId() {return id;}// 获取父节点IDpublic String getParentId() {return parentId;}// 获取节点名称public String getName() {return name;}
}// 主程序
public class Main {public static void main(String[] args) {// 创建实体列表,模拟树形结构List<Entity> entities = Arrays.asList(new Entity("1", null, "Root"), // 根节点new Entity("2", "1", "Child 1"), // Child 1, 父节点为Rootnew Entity("3", "1", "Child 2"), // Child 2, 父节点为Rootnew Entity("4", "2", "Grandchild 1") // Grandchild 1, 父节点为Child 1);// 创建树构建器实例TreeBuilder<Entity> treeBuilder = new TreeBuilder<>();// 构建树,传入实体列表及提取ID和父ID的函数TreeNode<Entity> tree = treeBuilder.buildTree(entities,Entity::getId, // 提取IDEntity::getParentId // 提取父ID);// 打印树结构printTree(tree, 0);}// 递归打印树结构的辅助方法private static void printTree(TreeNode<Entity> node, int level) {if (node == null) return; // 如果节点为空,直接返回// 打印当前节点的名称,缩进显示层级System.out.println(" ".repeat(level) + node.getData().getName());// 递归打印所有子节点for (TreeNode<Entity> child : node.getChildren()) {printTree(child, level + 1);}}
}
代码注释说明:
- Entity类:每个字段和方法都有注释,说明其用途和功能。
- Main类:
- 实体列表中每个节点的构造都有注释,解释其层级关系。
- 构建树的步骤清晰描述,便于理解。
- printTree方法:详细说明其递归逻辑和输出格式。