对于统计一个文件夹下面的文件的数量,大家第一反应肯定是递归调用来实现,现在有这么一个目录结构:root1和root2下面各有一个文件file1.txt和file2.txt,所以最终统计出来的文件数量应该是3

我们先看看递归实现:
- public static void countFiles(File file) {
- File[] files = file.listFiles();
- for (File file1 : files) {
- if (file1.isFile()) {
- fileCount++;
- } else {
- countFiles(file1);
- }
- }
- }
-
- @Test
- public void testCountFiles() {
- File root = new File("E:\\root");
- countFiles(root);
- System.err.println(fileCount);
- }
那么除了递归的方式,还有没有其他方法呢,答案是肯定的,我们知道递归的本质其实是在jvm的栈上分配方法栈帧,而jdk恰到好处的为我们提供了栈的实现,也就是java.util.Stack,那么就可以利用它来实现文件数量的统计了,直接看代码:
- public static int countFile(File file) {
- int count = 0;
- Stack<File> stack = new Stack<>();
- stack.add(file);
- while (!stack.isEmpty()) {
- File pop = stack.pop();
- if (pop.isFile()) {
- count++;
- }
- if (pop.isDirectory()) {
- File[] files = pop.listFiles();
- for (File file1 : files) {
- stack.add(file1);
- }
- }
- }
- return count;
- }
-
- @Test
- public void testStack() {
- File root = new File("E:\\root");
- System.err.println(countFile(root));
- }
ps:其实这里不一定要用Stack,也可以用queue等集合都可以实现,与递归的方式相比,各有优缺点吧,如果文件夹的数量过多,递归可能会导致StackOverflow,第二种方式可能会导致oom