算法导论 第8章 线性时间排序-计数排序的原址排序

//FILE:LINEAR_SORT.CPP
//实现CLRS第8章计数排序
//实现计数排序的原址排序
#include <iostream>
#include <vector>

using namespace std;
//实现CLRS第8章计数排序
void counting_sort(const vector<int>& a,vector<int>& b,const int k)
{
	if(a.size()!=b.size())
		cout<<"error"<<endl;
	vector<int> c(k+1,0);
	for (int i=0;i<a.size();i++)
		c[a[i]]++;
	for (int j=1;j<=k;j++)
		c[j]+=c[j-1];
	for (int m=a.size()-1;m>=0;m--)
		b[--c[a[m]]]=a[m];
}
//实现计数排序的原址排序
void counting_sort(vector<int>& A,const int k)
{
	vector<int> C(k+1,0);
	for (int i=0;i<A.size();i++)
		C[A[i]]++;
	for (int j=1;j<=k;j++)
		C[j]+=C[j-1];
	//将A初始化为0向量。
	//之后无需插入0元素
	for (int i=0;i<A.size();i++)
		A[i]=0;
	//根据C中计数从后向前插入数据
	//如果C[l]==C[l-1]，说明l已经插入完毕或A中本来就没有l元素
	for (int l=k;l>=1;l--)
		while (C[l]!=C[l-1])
			//这里C[l]--减去1是因为数组下标从0开始
			A[(C[l]--)-1]=l;
}
void main()
{
	int arr[]={2,5,3,0,2,3,0,3};
	vector<int> A(arr,arr+sizeof(arr)/sizeof(int));
	vector<int> B(A.size(),0);
	int k=5;
	//实现CLRS第8章计数排序
	counting_sort(A,B,k);
	for(int i=0;i<B.size();i++)
		cout<<B[i]<<" ";
	cout<<endl;
	//实现计数排序的原址排序
	counting_sort(A,k);
	for(int i=0;i<A.size();i++)
		cout<<A[i]<<" ";
	cout<<endl;
}