一道算法题——降低空间复杂度的思考

最近在看算法（第四版），里面有一道算法题。第一次写的时候，采用了双层嵌套，时间复杂度为o(n^2); 后来发现时间复杂度可以降低到o(n).

原题：编写一个静态方法histogram(),接受一个整型数组a[]和一个整数M为参数并返回一个大小为M的数组，其中第i个元素的值为整数i在参数数组中出现的次数。如果a[]中的值均在0到M-1之间，返回数组中所有元素之和应该和a.length相等。

最后一句话本来就值得玩味，跟a.length相等这个就要想一会儿，还真相等。所有项都被计数了，自然相等呗。

第一次写的复杂度o(n^2)的代码块：

public static int[] histogram(int[] a, int M) {
        int[] b = new int[M];
        for (int i = 0; i < a.length; i++) {
            int j = 0;
            for (int element : a) {
                if (element == i)
                    j++;
            }
            b[i] = j;
        }
        return b;
    }

还专门设置了j来计数，计数了某个数出现的次数，在写入到返回的数组中。两个for嵌套了。

改进后的代码如下：

public static int[] histogram(int[] a, int M)
    {
        int[] h = new int[M];
        int N = a.length;

        for (int i = 0; i < N; i++)
            if (a[i] < M)
                h[a[i]]++;

        return h;
    }

这样直接把某个数所在项加一，只用循环一次就好，大大降低时间复杂度。