第2章 排序 | 第10节 计数排序练习题 && 基数排序

对于一个int数组，请编写一个计数排序算法，对数组元素排序。

给定一个int数组A及数组的大小n，请返回排序后的数组。

测试样例：

[1,2,3,5,2,3],6

[1,2,2,3,3,5]

• 计数排序

class CountingSort {
public:
    int* countingSort(int* A, int n) {
        // write code here
        int min=A[],max=A[];
        for(int i=;i<n;i++)
        {
            if(A[i]<min) min=A[i];
            if(A[i]>max) max=A[i];
        }
        int k=max-min+;
        int* B=new int[k](); //初始化为0
        for(int i=;i<n;i++)
            B[A[i]-min]++;
        int idx=;
        for(int i=min;i<=max;i++)
            for(int j=;j<B[i-min];j++)
                A[idx++]=i;
        delete []B;
        return A;
    }
};

对于一个int数组，请编写一个基数排序算法，对数组元素排序。

给定一个int数组A及数组的大小n，请返回排序后的数组。保证元素均小于等于2000。

测试样例：

[1,2,3,5,2,3],6

[1,2,2,3,3,5]

• 基数排序

#include <iostream>
using namespace std;
const int MAX = ;

void print(int *a, int sz) {
    for (int i = ; i < sz; i++)
        cout << a[i] << " ";
    cout << endl;
}

void RadixSortLSD(int *a, int arraySize)
{
    int i, bucket[MAX], maxVal = , digitPosition = ;
    for (i = ; i < arraySize; i++) {
        if (a[i] > maxVal) maxVal = a[i];
    }

    int pass = ;  // used to show the progress
    /* maxVal: this variable decide the while-loop count
    if maxVal is 3 digits, then we loop through 3 times */
    while (maxVal / digitPosition > ) {
        /* reset counter */
        int digitCount[] = {  }; //对每一位：统计0-10对于数字的个数

        /* count pos-th digits (keys) */
        for (i = ; i < arraySize; i++)
            digitCount[a[i] / digitPosition % ]++;

        /* accumulated count */
        for (i = ; i < ; i++)
            digitCount[i] += digitCount[i - ];

        /* To keep the order, start from back side */
        for (i = arraySize - ; i >= ; i--)
        {
            int temp = a[i] / digitPosition % ;
            bucket[--digitCount[temp]] = a[i];
        }

        /* rearrange the original array using elements in the bucket */
        for (i = ; i < arraySize; i++)
            a[i] = bucket[i];

        /* at this point, a array is sorted by digitPosition-th digit */
        cout << "pass #" << pass++ << ": ";
        print(a, arraySize);

        /* move up the digit position */
        digitPosition *= ;
    }
}