基数排序

Posted 2020-07-06

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

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

测试样例：

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

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

代码：

class RadixSort {
public:
    int get(int a, int exp) {
        int b = a / (pow(10,exp));
        return b % 10;
    }
     
    int* radixSort(int* A, int n) {
        int maxd=0;
        for(int it=0; it<n; it++){
            if(A[it]>maxd)
                maxd=A[it];
        }
        int time=0;
        while(maxd){
            time++;
            maxd/=10;
        }               //得到最大的数有几位  

        queue<int> B[10];
        for(int i=0; i<time; i++){
            for(int j=0; j<n; j++){
                int x=get(A[j], i);
                B[x].push(A[j]);
            } 
            int index=0;
            for(int k=0; k<10; k++){
                for(unsigned int t=0; t<B[k].size(); ){
                    A[index++]=B[k].front();
                    B[k].pop();
                }
            }
        }
        return A;
    }
};