java - 使用堆栈在 Java 中制作文件/目录树

Question

我需要使用堆栈数据结构返回或打印文件/目录树。例如:

Folder1
    Folder1.1
        File1.1.1
    Folder1.2
Folder2
    File2.1
    File2.2
...

到目前为止我的代码:

public static void filetree(File mainfolder, int indent){

    Stack<String> filesanddirectories = new Stack<>();

        for (File file : mainfolder.listFiles()){
            if (file.isDirectory()){
                filesanddirectories.push(file.getName());
                filetree(file, 0);
            }
            else if (file.isFile()){
                filesanddirectories.push(file.getName());
            }
        }

    for (int i = 0; i < filesanddirectories.size(); i++){
        System.out.println(filesanddirectories.pop());
    }
}

此代码打印文件夹和文件，但没有缩进、向后且顺序不完全正确。

谁能解释一下它应该如何工作的逻辑？

编辑:找到了一个使用递归和堆栈的解决方案(尽管堆栈似乎是不必要的)

Answer 1

我使用 indent 和 temp 变量在递归遍历文件目录时保持缩进。然后在 for 循环的开头打印缩进。

public static void filetree(File mainfolder, int indent) {

        int temp;

        for (File file : mainfolder.listFiles()) {
            for(int i = 0; i<indent; i++) {
                System.out.print("  ");
            }
            temp = indent;
            if (file.isDirectory()) {

                indent++;
                System.out.println(file.getName()); 

                filetree(file, indent);
                indent--;

            } else if (file.isFile()) {

                System.out.println(file.getName());
                indent = temp;
            } 
        }
    }