《算法导论》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)

for i ← 0 to k
do C[i] ← 0
for j ← 1 to length[A]
do C[A[j]] ← C[A[j]] + 1
for i ← 1 to k
do C[i] ← C[i] + C[i - 1]
for j ← length[A] downto 1
do B[C[A[j]]] ← A[j]
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;
}