计数排序(Counting Sort)
一、概念
- 用空间换时间,在某些时候,平均时间复杂度可以比
O(nlogn)
更低。 - 计数排序的思想是,统计每个整数在序列中出现的次数,进而推导出每个整数在有序序列中的索引。
- 例如有如下一组数:
image
- 统计每个数出现的次数:
image
- 接着就可以顺序输出获得排序后的结果:
image
- 代码实现如下:
image
- 这个版本的时间复杂度是
O(n)
。 - 这个版本存在以下问题:
- 无法对负整数进行排序。
- 极其浪费内存空间。
- 不稳定的排序。
二、改进思路
- 有如下一组数,最小值为
3
,最大值为8
:
image
- 我们只需要一个数组,能存放
3-8
之间的数即可,则实际需要的数组空间为:max - min + 1
。
image
- 将每个元素出现的次数累加上其前面的所有次数,得到的就是元素在有序序列中的位置信息。
image
- 元素
8
获得的累计次数为8
,前一个元素7
出现的次数是7
,那么可以知道元素8
在数组中出现的次数为1
,排序后的索引为7
。
image
- 总结:
image
三、实际操作
- 首先,拥有
蓝色数组array
,即可获得绿色counts数组
,为了排序我们再新建一个紫色数组
。
image
- 从后往前遍历数组,首先是元素
5
,元素5
在紫色数组中的索引应该是4-1 = 3
,所以将5
插入数组索引为3
的位置,并且将元素5
的次数减1
。
image
- 接着遍历元素
4
,其在紫色数组中的索引为1
。
image
- 按照以上逻辑,依次遍历蓝色数组剩下的元素:
image
image
image
image
- 我们发现,从后往前遍历数组,可以保证不改变相同元素在新数组中的前后顺序。
四、代码实现
protected void sort() {
// 找出最值
int max = array[0];
int min = array[0];
for (int i = 1; i < array.length; i++) {
if (array[i] > max) {
max = array[I];
}
if (array[i] < min) {
min = array[I];
}
}
// 开辟内存空间,存储次数
int[] counts = new int[max - min + 1];
// 统计每个整数出现的次数
for (int i = 0; i < array.length; i++) {
counts[array[i] - min]++;
}
// 累加次数
for (int i = 1; i < counts.length; i++) {
counts[i] += counts[i - 1];
}
// 从后往前遍历元素,将它放到有序数组中的合适位置
int[] newArray = new int[array.length];
for (int i = array.length - 1; i >= 0; i--) {
newArray[--counts[array[i] - min]] = array[I];
}
// 将有序数组赋值到array
for (int i = 0; i < newArray.length; i++) {
array[i] = newArray[I];
}
}
复制代码
- 最好,最坏,平均时间复杂度为
O(n + k)
- 空间复杂度为
O(n + k)
。 -
k
是整数的取值范围。 - 属于
稳定排序
。