java计算参数目录大小的递归和迭代实现

练习题目代码如下:

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;

/**
 * Created by longforus on 8:50 PM 5/11/2016 .
 * IDEA-Chapter20_28 .
 * 把目录或文件作为参数传入 求其大小
 */


public class Chapter20_28 {
    public static void main (String[] args) throws IOException {
        System.out.println (getFormatSize (getSize (new File (args[0]))));
    }


//    public static long getSize (File file) {//递归求文件夹的大小
//        long size = 0;
//        if (file.isDirectory ()) {
//            File[] files = file.listFiles ();
//            for (int i = 0; i < files.length; i++) {
//                size += getSize (files[i]);
//            }
//        } else {
//            size += file.length ();
//        }
//        return size;
//    }

    public static long getSize (File file) {//迭代实现
        long size = 0;
        ArrayList<File> list = new ArrayList<> ();//运用了栈的原理
        if (file.isDirectory ()) {
            addToList (list, file);//是目录就把子目录入栈
        } else {
            size += file.length ();//是文件就加入总size内
        }
        while (! list.isEmpty ()) {//只要栈不空
            File temp = list.remove (list.size () - 1);//取出栈顶处理
            if (temp.isDirectory ()) {
                addToList (list, temp);
            } else {
                size += temp.length ();
            }
        }
        return size;
    }

    public static void addToList (ArrayList<File> list, File file) {//把file的内容入栈
        File[] files = file.listFiles ();
        for (int i = 0; i < files.length; i++) {
            list.add (files[i]);
        }
    }
    public static String  getFormatSize(long size){//得到格式化字符串
        double tempSize = size;
            int n = 0;
        String unit = "Byte";
        while (tempSize > 1024) {
            tempSize/=1024;
            n++;
        }
        switch (n) {
            case 1:
                unit = "KByte";
                break;
            case 2:
                unit = "MByte";
                break;
            case 3:
                unit = "GByte";
                break;
        }
            return  String.format ("%.2f%s",tempSize,unit);
    }
}

在这个例子中,递归实现确实比迭代实现要简单一些,

在自己实现迭代的时候,起先想到用树型结构,但是后面觉得空间的复杂度和遍历方面不是很理想,

后来想了一下想到了这个方法用栈来实现确实是比较理想的

应该还有更好的实现方法,原谅我还没有想到,

因为这个例子让我想了好久,才想到这个方法所以记在這里

经验还是很欠缺,脑子也不行啊