计数排序

Posted 2020-07-06

对于一个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) {
        int min=A[0], max=A[0];
        for(int i=1; i<n; i++){   //找桶的数量
            if(A[i]>max) max=A[i];
            if(A[i]<min) min=A[i];
        }
        int cnt=max-min+1;
        vector<int> b(cnt, 0);    //初始化桶
        for(int j=0; j<n; j++){   //装桶
            b[A[j]-min]++;        //A[]中的数对应的桶中的数量++
        }
        int it=0;   //A的下标
        for(int k=0; k<cnt; k++){
            while(b[k]-- > 0)    //b(桶)中为该数据出现的次数
                A[it++]=k+min;   //数据 = b的下标 + min
        }
        return A;
    }
};