转自算法导论第二版8.2节
计数排序不是比较排序。计数排序是用了输入元素的实际值来确定它们在数组中的位置。
计数排序假设n个输入数据中的每一个都是介于0到k之间的整数,此外k为某个整数。当k=O(n)时,计数排序的运算时间为Θ(n)。
注意这个算法的前提是每个数据都是0<= xi <= k <= n的整数。
计数排序的基本思想就是对每个输入元素x,确定出小于x的元素个数。有了这个信息,就可以把x直接放到他在最终输出数组中的位置。
在计数排序算法的代码中,我们假定输入时个数组A[1...n],length[A] = n。另外还需要两个数组:存放排序结果的B[1...n],以及提供临时存储区的C[0...k]。
计数排序的时间代价是Θ(n+k)。在实践中,当k=O(n)时,我们常常采用计数排序,这样其运行时间为Θ(n)。
计数排序的一个重要性质就是它是稳定的:具有相同值的元素在输入数组中的相对次序与他们在输入数组中的次序相同。亦即,两个相同数之间的顺序是这样来规定的,即在输入数组中先出现的,在输出数组中也位于前面。
伪代码
COUNTING-SORT(A, B, k)
for i <- 0 to k
do C[i] <- 0
for j <- 1 to length[A]
do C[A[j]] <- C[A[j]] +1
for i <- 1 to k
do C[i] <- C[i] + C[i-1]
for j <- length[A] downto 1
do B[C[A[j]]] <- A[j]
C[A[j]] <- C[A[j]] -1
#include <stdio.h>
void countingsort(int a[], int b[], int len, int k)
{
int i;
int c[k+1];
for (i = 0; i <= k; i++) {
c[i] = 0;
}
for (i = 1; i <= len; i++ ) {
c[a[i]]++;
}
for (i = 1; i <= k; i++) {
c[i] += c[i-1];
}
for (i = len; i>= 1; i--) {
b[c[a[i]]] = a[i];
c[a[i]]--;
}
}
int main(void)
{
int i;
int A[] = {
0, 2, 5, 3, 0, 2, 3, 0, 3
};
int B[9];
countingsort(A, B, 8, 5);
for (i = 1; i < 9; i++) {
printf("%d ", B[i]);
}
return 0;
}