java 生成目录树

废话不说，直接上代码：

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Stack;

public class FileTreeTest {
    
    public static void main(String[] args) {
        FileTree tree = new FileTree("f:/TDDOWNLOAD");
        tree.tree();        
    }
}

class FileTree {
    
    private File root;
    
    public FileTree(File root) {
        this.root = root;
    }
    
    public FileTree(String rootName) {
        this(new File(rootName));
    }
    
    public void tree() {
        Stack<Boolean> lastStack = new Stack<Boolean>();
        tree(root, 0, lastStack);
    }

    private void tree(File dir, int level, Stack<Boolean> lastStack) {
        if(dir == null || !dir.exists()) {
            return;
        }
        //  输出
        printFile(dir, level, lastStack);
        
        if(dir.isFile()) {            
            return;
        }
        File[] children = dir.listFiles();
        for(int i = 0; i < children.length; i++) {
            lastStack.push(i == children.length - 1);
            tree(children[i], level + 1, lastStack);
            lastStack.pop();
        }
    }
    
    /**
     * 打印树形文件目录
     * @param file 文件
     * @param level 文件所处的层级
     * @param lastStack  该文件是否是最后一个文件
     */
    private void printFile(File file, int level, Stack<Boolean> lastStack) {
        for(int i = 0, k = lastStack.size() - 1; i < k; i++) {
            System.out.print(lastStack.get(i) ? "\t" : "|\t");
        }        
        if(level > 0) {            
            System.out.print(lastStack.get(lastStack.size() - 1) ? "└ " : "|--");
        }
        System.out.println(level == 0 ? file.getPath() : file.getName());        
    }
    
}

最后输出类似于这种格式的。