根据要排序的元素的个数创建一个数组,数组的下标为0到n,然后数组的初始化的值为0,然后根据元素的个数进行计数,每多一个数,数组对应下标里的值加1,最后循环输出数组,此时数组里的元素就是有序的它的时间复杂度小于O(nlogn)`
package shujujiegou;
import java.util.Arrays;
public class CountSort {
public static int[] countSort(int[] array) {
//得到数组的最大值
int max=array[0];
for(int i=1;i<array.length-1;i++) {
if(array[i]>max) {
max=array[i];
}
}
//根据数组的最大值统计数组的长度 max表示新数组的下标
int[] countArray=new int[max+1];
//遍历数组,填充统计数组
for(int i=0;i<array.length;i++) {
countArray[array[i]]++;
}
//遍历统计数组,输出结果
int index=0;
int[] sortArray=new int[array.length];
for(int i=0;i<countArray.length;i++) {
for(int j=0;j<countArray[i];j++) {
sortArray[index++]=i;
}
}
return sortArray;
}
//优化后的计数排序
public static int[] countSort1(int[] array) {
int max=array[0];
int min=array[0];
for(int i=0;i<array.length;i++) {
if(array[i]>max) {
max=array[i];
}
}
for(int i=0;i<array.length;i++) {
if(array[i]<min) {
min=array[i];
}
}
int d=max-min;
int[] countArray=new int[d+1];
for(int i=0;i<array.length;i++) {
countArray[array[i]-min]++;
}
//统计数组做变形,后面的元素等于前面元素的和
int sum=0;
for(int i=0;i<countArray.length;i++) {
sum+=countArray[i];
countArray[i]=sum;
}
//倒序遍历原始数组,从统计数组中找出到正确位置,输出到结果数组
int[] sortedArray=new int[array.length];
for(int i=array.length-1;i>=0;i--) {
sortedArray[countArray[array[i]-min]-1]=array[i];
countArray[array[i]-min]--;
}
return sortedArray;
}
public static void main(String[] args) {
int[] array=new int[] {1,2,2,4,3,3,5,3,6,7,9,8};
int[] sortArray=countSort(array);
System.out.println(Arrays.toString(sortArray));
}
}