算法面试题总结之深度优先遍历

最新推荐文章于 2025-12-27 16:49:17 发布

原创最新推荐文章于 2025-12-27 16:49:17 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

该代码段展示了一个Java函数，用于计算指定文件目录下的所有文件数量，包括隐藏文件但不包括子目录。它使用深度优先遍历策略，通过栈来存储待检查的文件或文件夹。当遍历到文件时，计数器增加；遇到目录，则将其添加到栈中继续遍历。在main方法中，函数被调用并打印出结果。

统计文件数量

题目描述：给定一个文件目录的路径,编写一个函数统计该目录下所有的文件数量并返回（隐藏文件也算,但文件夹不算）
解题思路：
使用深度优先遍历（遍历过程中选用栈作为存储文件或文件夹的容器），遍历之前，先将头文件压栈（栈是否为空作为遍历完所有文件的判断标准），如果当前路径上是文件，计数器加1，如果当前路径上是文件夹，则遍历文件夹中的内容，判断是文件还是文件夹

public class CountFiles {

    public static int countFileNum(String folderPath){
        File root = new File(folderPath);
        if (root.isFile()){
            return 1;
        }
        // 如果如上判断,没有返回结果,可知root不是文件,则root可能是文件夹
        if (!root.isDirectory() && !root.isFile()){
            return 0;
        }
        // root是目录
        int count = 0;
        Stack<File> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()){
            File cur = stack.pop();
            for (File next : cur.listFiles()) {
                if (next.isFile()){
                    count++;
                }
                if (next.isDirectory()){
                    stack.push(next);
                }
            }
        }
        return count;
    }

    public static void main(String[] args) {
        System.out.println(countFileNum("D:\\study_note"));
        System.out.println(countFileNum("D:\\study_note\\HeiMa"));
    }
}