《算法导论》CLRS算法C++实现（五）P98 计数排序

第八章线性时间排序

8.2 计数排序

这个算法比较简单。计数排序的基本思想就是对每一个输入元素x，确定出小于x的元素个数。然后就可以把x直接放到它的最终输出数组的位置上。例如如果有17个元素小于x，则x旅游属于第18个输出位置。

假定输入的日企数组A[1..n]，length[A]=n。另外还需要两个数组：存放排序结果的B[1..n]，以及提供临时存储区的C[0..k]。

COUNTING-SORT(A, B, k)

1 for i ← 0 to k
2     do C[i] ← 0
3 for j ← 1 to length[A]
4     do C[A[j]] ← C[A[j]] + 1
5 for i ← 1 to k
6     do C[i] ← C[i] + C[i - 1]
7 for j ← length[A] downto 1
8     do B[C[A[j]]] ← A[j]
9     C[A[j]] ← C[A[j]] - 1

C++程序实现

#include <iostream>
#include <cstring>

using namespace std;

void countingSort(int* A, int* B, int lengthA, int k)
{
    int* C = new int[k + 1];
    memset(C, 0, (k + 1) * sizeof(int));
    for(int j = 0; j < lengthA; j++)
    {
        C[A[j]] = C[A[j]] + 1;
    }
    for(int i = 1; i < (k + 1); i++)
    {
        C[i] = C[i] + C[i - 1];
    }
    for(int j = lengthA - 1; j >= 0; j--)
    {
        B[C[A[j]] - 1] = A[j];
        C[A[j]] = C[A[j]] - 1;
    }
    delete[] C;
}

int main()
{
    int arrInitial[] = {2, 5, 3, 0, 2, 3, 0, 3};
    int* arrFinal = new int[8];
    countingSort(arrInitial, arrFinal, 8, 5);
    for(int i = 0; i < 8; i++)
    {
        cout << arrFinal[i] << " ";
    }
    cout << endl;
    return 0;
}