算法第四版练习1.1.15

问题

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

解答

这个问题需要使用双重循环，先利用if语句判断第i个元素是否与数组中的数相等，若相等则表示出现一次，从而累加其出现次数，遍历完整个数组后number的值就是第i个元素出现的次数，这是内层循环。外层循环的作用是把M个number值依次放进返回的数组中。这里的外层循环我定义的下标从1开始而不是从0开始，因为题干中描述为“第i个元素”，第i个元素在返回的数组中下标为i-1，这个细节值得注意。
完整代码如下⬇

public class TestFile {
    public static void histogram(int[] a,int M) {
        int[] result = new int[M];
        for(int i=1;i<=M;i++) {
            int number = 0;
            for (int j=0;j<a.length;j++) {
                if (i == a[j]) {
                    number ++;
                }
            }
            result[i-1] = number;
        }
        for (int k=0;k<M;k++) {
            System.out.printf("%d ",result[k]);
        }
    }

    public static void main(String[] args) {
        histogram(new int[] {1,3,5,2,2,1,4,1},8);
    }
}

测试代码

如main函数中一样，使用的测试数组为{1,3,5,2,2,1,4,1}，返回的数组所有数相加等于8，则正确。