问题:假设待排序记录均为整数且取自区间[ 0 , k ],计数排序(Count Sort)的基本思想是对每一个记录 x ,确定小于 x 的记录个数,然后直接将 x 放在应该的位置。例如,小于 x 的记录个数是 10 ,则 x 就位于第 11 个位置。
算法:
1、设置 num[] 数组(当成hash表用),统计出来每个数字出现的次数,并放入相应的位置,例如 2 出现了 5 次,那么num[2] = 5 。
2、统计小于等于 i 的记录个数,并覆盖原来的num[]数组。
3、反向填充B[]数组。多设置一个数组 B[] 的原因如下,如果不设置此数组,由num[]直接输出,例如 num[3] = 4 ,即有 4 个 3 ,输出则为 3 3 3 3 ,这时就不知道此时的3对应原数组中的哪个 3 ,不具有稳定性。设置之后,B[] 数组存放待排序数组 A[] 中元素排序后应在的位置,此时就知道待排序数组中第一个 3 排在哪里了,所以计数排序具有稳定性。
4、结果存放在result[]数组。
代码如下:
void CountSort(int A[],int n,int k,int B[]){
int i , num[k+1] = {0};
for(i=0;i<n;i++) num[A[i]++]++;//hash表计数
for(i=1;i<=k;i++) num[i] += num[i-1];//累计起来
for(i=n-1;i>=0;i--) B[i] = --num[A[i]-1];//-1是因为数组下标从0开始
}
数组下标 :0,1,2,3,4,5,6,7,8,9
A[ ] :2,1,5,2,4,3,0,5,3,2
num[ ] :1,1,3,2,1,2 //有1个0,1个1,3个2,2个3,1个4,2个5
num[ ] :1,2,5,7,8,10 //统计小于等于 i 的记录个数
B[ ] :2,1,8,2,7,5,0,9,6,4
所以遍历 A[] ,A[0]应该放在B[0]的位置上,即result[ B[0] ] = A[0],
A[1]应该放在B[1]的位置上,即result[ B[1] ] = A[1]...
检验如下:
int main(){
int A[10] = {2,1,5,2,4,3,0,5,3,2};
int B[10];
cout<<" 待排序数组为:";
for(int i=0;i<10;i++) cout<<A[i]<<" ";cout<<endl;
CountSort(A,10,5,B);
cout<<"计数排序输出的序号数组为:";
for(int i=0;i<10;i++) cout<<B[i]<<" ";cout<<endl;
int result[10] = {0};
for(int i=0;i<10;i++) result[B[i]] = A[i];
cout<<" 计数排序后的输出数组为:";
for(int i=0;i<10;i++) cout<<result[i]<<" ";cout<<endl;
return 0;
}
输出如下:
待排序数组为:2 1 5 2 4 3 0 5 3 2
计数排序输出的序号数组为:2 1 8 3 7 5 0 9 6 4
计数排序后的输出数组为:1 2 3 3 3 4 4 5 6 6