一、算法原理
计数排序(Counting sort)的基本思想是:使用一个辅助数组C和B,其中C中第i个元素是待排序数组A中值为i的元素的个数,B是排序后的数组,对每一个输入元素x,确定小于x的元素的个数,利用这一信息,就可以直接把x放在它在输出数组中的正确位置上了。例如,如果有15个元素小于x,那么x就应该在第16个输出位置上。
二、算法描述
假设规模为n的数组中每个元素的值都是在[0…k]区间内的一个整数.
- 首先,统计数组中每个值为i的元素出现的次数,存放到C[i]
- 然后,对所有的计数累加,从C中第二个元素开始,每一项的值等于该项和前一项相加的值
- 最后,反向填充排序数组,将每个元素i放在新数组的第C[i]项
三、算法复杂度
- 时间复杂度为 T(n) = O(n+k),实际工作中,当k=O(n)时,一般采用计数排序,这时运行时间就是T(n) = O(n)
- 空间复杂度,需要辅助空间O(n+k)
四、代码实现
//计数排序,长度为length的数组A的值在0~k范围内,分配存放排序后的数组B并返回
int * CountingSort(int A[], int length, int k)
{
//初始化B和C数组
int * B = (int *)malloc(length * sizeof(int));
for(int i = 0; i < length; i++)
B[i] = 0;
int C[k+1];
for(int i = 0; i <= k; i++)
{
C[i] = 0;
}
//C数组对A数组中值进行计数
for(int j = 0; j < length; j++)
{
C[A[j]]++;
}
printf("辅助计数数组C: \n");
for(int i = 0; i <= k; i++)
printf("%d : %d\n", i, C[i]);
printf("\n");
//将C数组计数次数每一项与前一项累加
for(int i = 1; i <= k; i++)
{
C[i] = C[i-1] + C[i];
}
printf("辅助计数数组C: \n");
for(int i = 0; i <= k; i++)
printf("%d : %d\n", i, C[i]);
printf("\n");
//在B数组中保存排序好的数组
for(int j = length - 1; j >= 0; j--)
{
B[C[A[j]]-1] = A[j];
C[A[j]]--;
}
return B;
}