计数排序的基本思想:对每一个输入元素 x ,确定小于 x 的元素个数。利用这一信息,就可以直接把 x 放在输出数组中对应的位置上。例如,假如有 17 个元素小于 x,则 x 就应该放在第 18 个输出位置上。当数组中有相同的元素时,要略微修改这个方案,因为两个元素不能放在同一个位置上。
我们把计数排序称为非比较排序,因为计数排序的实现完全不依赖于任何元素之间的大小关系。与计数排序相比,冒泡排序则是一种典型的比较排序。
计数排序牺牲了空间复杂度,它必须在内存中开辟新的空间来存放排序后的值,并且返回指向这个数组的指针,也因此它无法做到原址排序。但是,计数排序的时间复杂度为O(n),与其他任何比较排序都优秀得多。
// 计数排序
#include <stdio.h>
#include <stdlib.h>
#define length(a) ((sizeof(a))/(sizeof(a[0])))
void printArray(int* a, int length)
{
int temp;
for (temp = 0; temp < length; temp++)
{
printf("%d ", a[temp]);
}
printf("\n");
}
int* countingSort(int* a, int max, int length);
int findMax(int* a, int length);
int main(void)
{
int a[] = { 5,2,4,6,1,3,6,10,15,9 };
printArray(a, length(a));
printArray(countingSort(a, findMax(a, length(a)), length(a)), length(a));
return 0;
}
int* countingSort(int* a, int max, int length)
{
int* b = (int*)malloc(sizeof(int) * length);
int* c = (int*)malloc(sizeof(int) * max);
for (int i = 0; i < max; i++)
c[i] = 0;
for (int i = 0; i < length; i++)
c[a[i] - 1]++;
for (int i = 1; i < max; i++)
c[i] += c[i - 1];
for (int i = length - 1; i >= 0; i--)
{
b[c[a[i] - 1] - 1] = a[i];
c[a[i] - 1]--;
}
free(c);
return b;
}
int findMax(int* a, int length)
{
int max = a[0];
for (int i = 0; i < length; i++)
{
if (a[i] > max)
max = a[i];
}
return max;
}