【数据结构】计数排序

对于一个int数组，请编写一个计数排序算法，对数组元素排序。
给定一个int数组A及数组的大小n，请返回排序后的数组。
测试样例：
[1,2,3,5,3],6
[1,5]

#include<iostream>
#include<string>
using namespace std;


class CountingSort {
public:

    void display(int * A,int n){
        for (int i = 0; i < n; i++){
            printf("%d ",A[i]);
        }
        printf("n");
    }
    void swap(int * A,int * B){
        int temp = *A;
        *A = *B;
        *B = temp;
    }


    int* countingSort(int* A,int n) {
        // write code here

        int min_val,max_val;
        min_val = max_val = A[0];
        for (int i = 0; i < n; i++){
            if (A[i] < min_val)
                min_val = A[i];
            if (A[i] > max_val)
                max_val = A[i];
        }

        int n1 = max_val - min_val + 1;
        int * a = new int[n1]();

        for (int i = 0; i < n; i++){
            a[A[i] - min_val]++;
        }

        int counter = 0;
        for (int i = 0; i < n1; i++){
            for (int j = 0; j < a[i]; j++){
                A[counter++] = i + min_val;
            }
        }

        delete[] a;

        return A;
    }
};


int main()
{
    CountingSort s;
    int A[] = { 54,35,48,36,27,12,44,8,14,26,17,28 };
    int n = 13;
    s.countingSort(A,n);
    s.display(A,n);
}