给定一个文件夹,不允许用递归,统计其下面的文件数量,包括子文件夹下面的文件

对于统计一个文件夹下面的文件的数量,大家第一反应肯定是递归调用来实现,现在有这么一个目录结构: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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值