计数排序--C++实现

1. 描述

对于一列在范围在[a,b]数，构建一个数组arr，
数组的大小为 b - a + 1。

使得任意该范围内的数都能一一对应到创建的数组中。对于待排序的数列S,对数列中的每一个数，在构建的数组中进行计数(S[arr[i]]++)。最后从小到大遍历该数组arr,输出arr[i]个对应的数。

2. 实现

#include <algorithm>
#include <iostream>
#include <cstring>

void countSort( int *arr, int sz, int *output)
{
    if (!arr || !sz || !output)
        return ;


    int minVal = arr[0];
    int maxVal = arr[0];

    for ( int i = 1; i < sz; ++i) {
        minVal = std::min(minVal, arr[i] );
        maxVal = std::max(maxVal, arr[i] );
    }

    std::cout << minVal << " " << maxVal << std::endl;

    int bucketSz = maxVal - minVal + 1;
    int *bucket = new int[bucketSz];
    memset(bucket, 0, sizeof(int) * bucketSz);


    for ( int i = 0; i < sz; ++i )
        bucket[ arr[i] - minVal ]++;

    for ( int i = 0,cnt = 0; i < bucketSz; ++i ) {
        while ( bucket[i] ) {
            output[cnt++] = minVal + i;
            bucket[i]--;
        }
    }

    delete[] bucket;
}


int main (int argc, char *argv[])
{

    int arr[10] = { -2,1,6,9,0,-10,2,3,6,4};
    int output[10] = {0};
    int sz = 10;

    countSort( arr, sz, output );

    for (int i = 0; i < sz; ++i ) {
        std::cout << " " << output[i];
    }

    return 0;
}

3. 分析

可以看到，计数排序需要事先知道数据的大致范围，否则需要遍历一遍取得数据的范围。同时需要O(maxVal - minVal + 1)的额外空间来保存每个值的计数；若实现知道数据的范围，则需要遍历一遍待排序数组和计数数组，空间消耗为O(arr.size + valRange.size)