计数排序不是比较排序,排序速度快于任何比较排序算法,它是一种稳定的线性时间排序算法。
假设有n输入元素(均为整数)中的每个值都在区间[0,k)(k为整数)上,则辅助数组大小为k。辅助数组第i个元素表示待排序数组中小于等于i的数出现的次数。
以下是计数排序的实现,假设区间为[0,100)。
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
void display(const int *A, int len)
{
int i;
for (i = 0; i < len; i++) {
printf("%d ", A[i]);
}
printf("\n");
}
void counting_sort(const int *A, int *B, int len)
{
int *C = (int *)malloc(sizeof(int) * 100);
int i, j, k;
assert(C);
for (i = 0; i < 100; i++) {
C[i] = 0;
}
/* After this for loop, C[i] now contains the number
* of elements equal to i.
*/
for (i = 0; i < len; i++) {
C[A[i]]++;
}
/* After this for loop, C[i] now contains the number
* of elements less than or equal to i.
*/
for (i = 1; i < 100; i++) {
C[i] += C[i - 1];
}
for (i = len; i > 0; i--) {
B[--C[A[i - 1]]] = A[i - 1];
}
free(C);
}
int main()
{
int *A, *B;
int i, n = 10;
A = (int *)malloc(sizeof(int) * n);
B = (int *)malloc(sizeof(int) * n);
assert(A && B);
srand(time(NULL));
for (i = 0; i < n; i++) {
A[i] = rand() % 100;
}
display(A, n);
counting_sort(A, B, n);
display(B, n);
free(A);
free(B);
return 0;
}
在上述的实现中,若将
for (i = len; i > 0; i--) {
B[--C[A[i - 1]]] = A[i - 1];
}
修改为:
for (i = 0; i < len; i++) {
B[--C[A[i]]] = A[i];
}
这样修改后,相同的元素排列相对位置正好与原位序相反,这将导致计数排序算法不是稳定的。
计数排序在排序时需要知道待排序数组中的数据范围,这使得它需要大量的内存空间。
参考
[1]: 计数排序
[2]: 《算法导论》 8.2 计数排序