算法分析与设计之计数排序

计数排序通过两个数组存放，一个result数组用于装排序结果，一个count数组用于遍历并处理计数结果。Count[i]中i就是元素大小，count[i]的值就是该元素的个数。

import java.util.Arrays;
import java.util.Random;
public class CS {

    public static int[] countSort(int[] arr) {
        //求最大值，得出数组长度
        int max = 0;

        //遍历求最大值
        for (int i = 0; i < arr.length; i++) {
            max = Math.max(max, arr[i]);
        }
        int[] count = new int[max + 1];

        //遍历计数
        for (int i = 0; i < arr.length; i++) {
            count[arr[i]] ++;
        }

        //用作装排序结果的数组
        int[] result = new int[arr.length];

        //上述数组的记录下标
        int index = 0;

        //遍历计数的数组，并处理计数结果
        for (int i = 0; i < count.length; i++) {

            //遇到计了数的
            if (count[i] > 0){

                //重要：i的大小就是元素大小，count[i]就是该元素的个数
                //将i元素装count[i]次
                for (int elementNum = 0; elementNum < count[i]; elementNum++) {
                    result[index++] = i;
                }
            }
        }//计数排序统计完毕

        //返回计数排序完成的有序新数组
        return result;
    }

    /**
     * 验证计数排序算法
     * @param args
     */
    public static void main(String[] args) {
        int arr[] = new int[15];
        Random random = new Random();
        for (int i = 0; i < arr.length;i++){
            arr[i] = random.nextInt(10);
        }
        System.out.println("生成的随机数是："+ Arrays.toString(arr));

        //进行计数排序，返回排好序的新数组
        int[] countSortedArr = countSort(arr);

        //这里可以将排好序的数组，重新赋给原来的数组，保持之前的操作
        arr = countSortedArr;
        System.out.println("堆排序后：");
        //打印计数排序后的数组
        System.out.println(Arrays.toString(arr));
    }

}