具体的讲解参见:理解计数排序算法的原理和实现
主要的原理,就是用一个辅助数组 b
,对于目标数组 a
,对于 a 中的每个元素 a[i]
,将其出现的次数映射到 b[j]
中,其中有 j = a[i]
。
然后需要注意的三个点就是:
1、尽量的节省空间,假设有数组 10, 11, 20, 30
,辅助数组的大小没必要为 30,可以根据数组中的最大值与最小值来进行优化。
2、要支持负数的排序
3、具有稳定性,重复元素的相对位置要保持不变
具体的代码实现为:
public int[] countSort(int[] a) {
int[] b = new int[a.length];
int max = a[0], min = a[0];
for (int i : a) {
if (i > max) {
max = i;
}
if (i < min) {
min = i;
}
}
// 根据 a 中的最大值与最小值,求出辅助数组的最小长度
int[] c = new int[max - min + 1];
for (int i1 : a) {
c[i1 - min] += 1;
}
// 进行一次累加操作之后,
// 那么对于存在数个相等的元素 a[x](假设为 a[x]_1,a[x]_2,...a[x]_last,
// 其中 a[x]_last 为最后一个 a[x],即在 a 中处于最右边位置的 a[x])
// 则在排序后,有 (c[i] - 1) 即为 a[x]_last 前面存在的元素个数。
// 这样是保证排序结果稳定的前提
for (int i = 1; i < c.length; ++i) {
c[i] = c[i] + c[i - 1];
}
// 倒着遍历 a,则可以先遍历到 a[x]_last,从而根据 c[y] 的值来确定其稳定的位置
for (int i = a.length - 1; i >= 0; --i) {
int pos = a[i] - min;// pos 为 a[i] 映射到 c 中的下标
int sumCount = c[pos];
b[sumCount - 1] = a[i];
--c[pos];
}
return b;
}
引用 理解计数排序算法的原理和实现 的原话:
第一,在于理解计算max和min之后,需要使用原数组每一个元素减去min的转换值统计词频,特定情况下能节省存储空间,这样做的另一个好处是可以兼容负数的情况,因为每一个元素减去最小值之后,结果必定是大于等于0
计数排序是一个非基于比较的排序算法,它的优势在于在对一定范围内的整数排序时,它的复杂度为 Ο(n+k)(其中k是整数的范围),快于任何比较排序算法。且空间复杂度也为 O(n+k)。
而且当 O(k)>O(n log(n)) 的时候其效率反而不如基于比较的排序(基于比较的排序的时间复杂度在理论上的下限是 O(n log(n) ), 如归并排序,堆排序)。