当前位置：首页 > news >正文

Java将文件目录转成树结构

news 文章来源：https://blog.csdn.net/zuihongyan518/article/details/138336501 2024/5/19 22:06:30

在实际开发中经常会遇到返回树形结构的场景，特别是在处理文件系统或者是文件管理系统中。下面就介绍一下怎么将文件路径转成需要的树形结构。

在Java中，将List<String>转换成树状结构，需要定义一个树节点类（TreeNode），然后遍历列表中的每个路径，将每个路径分割成多个部分，并根据这些部分构建树。下面是一个简单的实现示例：

首先，定义一个树节点类TreeNode：

    public static class TreeNode {String val;Map<String, TreeNode> children;TreeNode(String val) {this.val = val;this.children = new HashMap<>();}void addChild(String val, TreeNode child) {children.put(val, child);}TreeNode getChild(String val) {return children.get(val);}}

然后，可以使用以下封装的代码将List<String>转换成树状结构：

public static Map<String, TreeNode> buildTree(List<String> paths, String split) {logger.info("[字符串转Tree结构]开始...params paths:{}", paths);if (paths == null || paths.isEmpty()) {return Collections.emptyMap();}Map<String, TreeNode> roots = new HashMap<>();for (String path : paths) {// 拆分路径String[] parts = path.split(split);if (parts.length == 0) {continue;}String rootVal = parts[0];TreeNode root = roots.get(rootVal);// 不存在，则创建它if (root == null) {root = new TreeNode(rootVal);roots.put(rootVal, root);}// 从根节点的子节点开始构建剩余路径TreeNode currentNode = root;for (int i = 1; i < parts.length; i++) {String part = parts[i];TreeNode child = currentNode.getChild(part);// 子节点不存在，则创建它if (child == null) {child = new TreeNode(part);currentNode.addChild(part, child);}// 移动到子节点，继续构建路径currentNode = child;}}return roots;}

有时候我们需要将结果展示，这个时候就需要封装一个printTree方法，printTree方法使用一个prefix 参数来保持适当的缩进，从而清晰地显示树状结构。它首先打印当前节点的名称，然后递归地调用自身来打印每个子节点，每次递归调用时都增加缩进。

    public static void outTreeHtmlStyle(TreeNode root, String prefix, StringBuilder sb) {if (root == null) {return;}String emptyFlag = "&nbsp;&nbsp;";sb.append(emptyFlag).append(prefix).append(root.val).append("</br>");for (TreeNode child : root.children.values()) {outTreeHtmlStyle(child, prefix + "|" + emptyFlag, sb);}}

最后在HTML中展示的效果如下：

相关文章：