对于一个int数组,请编写一个基数排序算法,对数组元素排序。
给定一个int数组A及数组的大小n,请返回排序后的数组。保证元素均小于等于2000。
测试样例:
[1,2,3,5,2,3],6
[1,2,2,3,3,5]
import java.util.*;
public class RadixSort {
public int[] radixSort(int[] A, int n) {
// write code here
int[] count = new int[2001];
for(int i=0;i<n;i++){
count[A[i]]++;
}
int index =0;
for(int j=0;j<count.length;j++){
while(count[j]-->0){
A[index++] = j;
}
}
return A;
}
}