【刷题之路】计数排序

经典算法

class CountingSort {
public:
    int* countingSort(int* A, int n) {
        // write code here
        int max=A[0],min=A[0],i,j=0,diff;
        for(i=0;i<n;i++){ //寻找最大与最小值，确定数组大小
            if(max<A[i]) max=A[i];
            if(min>A[i]) min=A[i];
        }
        diff=max-min+1;
        vector<int> temp(diff,0); //根据diff确定数组大小
        for(i=0;i<n;i++){
            temp[A[i]-min]++; //遇到某一个数，减去min即可从零开始，计数
        }
        for(i=0;i<diff;i++){
            while(temp[i]>0){
                A[j++]=min+i;  //遍历A，根据计数重新生成数组
                temp[i]--;
            }
        }
        return A;
        
    }
};