计数排序的基本思想是对于给定的输入序列中的每一个元素x,确定该序列中值小于x的元素的个数。
一旦有了这个信息,就可以将x直接存放到最终的输出序列的正确位置上。
例如,如果输入序列中只有17个元素的值小于x的值,则x可以直接存放在输出序列的第18个位置上。当然,如果有多个元素具有相同的值时,我们不能将这些元素放在输出序列的同一个位置上,因此,上述方案还要作适当的修改。
Java代码实现此算法如下:
/**
* 计数排序
* @param a 待排序的数组
* @param k 数组中元素的大小不大于k
*/
public static int[] countingSort(int[] a, int k) {
int[] b = new int[a.length];
int[] c = new int[k + 1];
for (int i = 0; i < a.length; i++)
c[a[i]] += 1;
for (int i = 1; i < c.length; i++)
c[i] += c[i - 1];
for (int i = a.length - 1; i >= 0; i--) { // 这个下标应该从后向前遍历,否则的话算法就是不稳定的了
b[c[a[i]] - 1] = a[i]; // 由于下标从0开始,这里的b下标应该减1
c[a[i]] -= 1;
}
return b;
}