一、算法步骤
(1)从待排序序列 A 中找到最大值 max 和最小值 min;
(2)创建一个新数组 B,用来统计序列 A 中各元素出现的次数。
(3)最后遍历数组 B,输出结果即为排序的结果。
注:
- 若 A 中的元素都比较小,则创建新数组 B 的长度为(max + 1),此时数组 B 的下标 index 就是序列 A 中的元素;
- 若 A中的元素都比较大,则创建新数组 B 的长度为(max - min + 1),可以节省空间,此时数组 B 的下标 index 加上最小值(即index + min)才是序列 A 的元素。
Tips:
计数排序适合待排序的数字在一个很小的范围内,并且可以使用辅助内存的情形。比如,要对一个公司的几万名员工的年龄进行排序,我们就可以创建一个长度为 100 的数组(假设员工年龄的范围是 0 ~99),用来统计每个年龄出现的次数。最后再将统计的结果遍历输出即可。
二、举例分析
假设有一序列 2,1,1,2,2,3,计数排序过程如下:
创建一个新数组 B,长度为 4(max + 1),统计待排序序列中各元素出现的次数,如下表:
遍历数组 B,将结果依次输出得到有序序列 1,1,2,2,2,3。
三、代码实现
public int[] countingSort(int[] sourceArray) {
// 拷贝数组
int[] array = Arrays.copyOf(sourceArray, sourceArray.length);
// 找数组中的最大、最小值,以确定新创建数组的长度
int maxValue = array[0];
int minValue = array[0];
for (int i = 1; i < array.length; i++) {
if (maxValue < array[i])
maxValue = array[i];
if (minValue > array[i])
minValue = array[i];
}
// 遍历原数组,记录各元素出现的次数并存入bucket对应下标的位置
// 注意:数组bucket中index处的元素记录的是array中某元素出现的次数
int[] bucket = new int[maxValue - minValue + 1];
for(int value : array){
bucket[value - minValue]++;
}
// 遍历bucket,输出目标序列(遍历的方向,决定输出结果的排序方向)
int index = 0;
for (int i = bucket.length - 1; i >= 0; i--) {// 左边最大右边最小
while(bucket[i] > 0){
array[index++] = i + minValue;
bucket[i]--;
}
}
return array;
}