如何让Java类实现Unix tree命令的目录树打印效果？

问题分析 #

你的代码目前存在几个关键格式问题，导致输出和标准tree命令不一致：

• 根目录被错误添加了├──前缀
• 所有子项统一使用├──，未区分最后一个子项的└──
• 层级缩进仅用重复的──填充，没有根据父节点位置显示│来体现层级关联

修改后的完整代码 #

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class TreePrint {
    private final Directory root;

    public TreePrint(String path) {
        root = new Directory(path);
    }

    public void printTree() {
        // 根目录单独打印，不需要前缀
        System.out.println(colorTextToBlue(root.directoryName));
        // 遍历根目录的子项，传递初始前缀和是否为最后一个的标记
        for (int i = 0; i < root.fileList.size(); i++) {
            boolean isLast = (i == root.fileList.size() - 1);
            root.fileList.get(i).print(isLast, "");
        }
    }

    private interface FileSystemComponent {
        // 修改print方法，增加isLast和prefix参数
        void print(boolean isLast, String prefix);
    }

    private static class FileComponent implements FileSystemComponent {
        private final String fileName;

        public FileComponent(String name) {
            fileName = name;
        }

        @Override
        public void print(boolean isLast, String prefix) {
            // 根据是否为最后一个元素，选择对应的前缀符号
            String symbol = isLast ? "└── " : "├── ";
            System.out.println(prefix + symbol + fileName);
        }
    }

    private static class Directory implements FileSystemComponent {
        private final String directoryName;
        private final List<FileSystemComponent> fileList = new ArrayList<>();

        public Directory(String path) {
            File directory = new File(path);
            if (!directory.exists()) {
                throw new IllegalArgumentException("Invalid path, please try again");
            }

            directoryName = directory.getName();
            File[] components = directory.listFiles();

            if (null != components) {
                for (File file : components) {
                    if (file.isDirectory()) {
                        fileList.add(new Directory(file.getPath()));
                    } else {
                        fileList.add(new FileComponent(file.getName()));
                    }
                }
            }
        }

        @Override
        public void print(boolean isLast, String prefix) {
            // 打印当前目录的前缀和名称
            String symbol = isLast ? "└── " : "├── ";
            System.out.println(prefix + symbol + colorTextToBlue(directoryName));
            
            // 构建子项的前缀：如果当前是最后一个目录，后续前缀用空格；否则用│ 保持竖线连接
            String childPrefix = prefix + (isLast ? "    " : "│   ");
            
            // 遍历子项，逐个处理是否为最后一个元素
            for (int i = 0; i < fileList.size(); i++) {
                boolean childIsLast = (i == fileList.size() - 1);
                fileList.get(i).print(childIsLast, childPrefix);
            }
        }
    }

    private static String colorTextToBlue(String text) {
        final String blueBold = "\033[1;34m";
        final String reset = "\033[0m";
        return blueBold + text + reset;
    }
}

关键修改说明 #

1. 根目录单独处理：在printTree中直接打印根目录名称，避免给根目录添加多余的前缀符号
2. 增强print方法参数：添加isLast标记区分最后一个子项，添加prefix传递层级缩进的格式信息，用于构建tree命令中的竖线和空格结构
3. 动态生成前缀符号：根据isLast选择├──或└──，同时根据父节点是否为最后一个，决定子项前缀用│ （保持竖线连接）还是 （空格填充）
4. 移除冗余的printLevel方法：将前缀生成逻辑整合到各个组件的print方法中，逻辑更清晰

示例输出 #

假设根目录是quiz19，包含你列出的文件，输出会变成：

quiz19
├── makefile
├── quiz19.c
├── stack_min.c
└── stack_min.h

如果存在嵌套目录，比如quiz19下有subdir目录，输出会是：

quiz19
├── makefile
├── quiz19.c
├── stack_min.c
├── stack_min.h
└── subdir
    ├── file1.txt
    └── file2.txt

内容的提问来源于stack exchange，提问作者May Sohatchevzzki