算法导论学习笔记——8.2计数排序

8.2计数排序

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<long long,long long> pll;
const int inf = 0x3f3f3f3f;
const int maxn=100010;
ll A[maxn],B[maxn],C[maxn];
ll n,k;
void CountingSort(ll A[],ll B[],ll k)
{
	for(int i=0;i<=k;i++)
		C[i]=0;
	for(int i=1;i<=n;i++)
		C[A[i]]++;
	for(int i=1;i<=k;i++)
		C[i]+=C[i-1];
	for(int i=n;i>=1;i--)
	{
		B[C[A[i]]]=A[i];
		C[A[i]]=C[A[i]]-1;
	}
}
int main()
{
	scanf("%lld%lld",&n,&k);
	for(int i=1;i<=n;i++)
		cin>>A[i];
	CountingSort(A,B,k);
	for(ll i=1;i<=n;i++)
		cout<<B[i]<<' ';
    return 0;
}