计数排序不同于其他排序那样比较数组中数组的大小来进行排序,而是一种基于非比较的排序算法,计数排序是桶排序的一种思想。
本实例中写了两种方法,一种普通版本的的计数排序和一种优化过的计数排序,注释已经写在代码里面了,比较简单。
图示过程:
示例代码:
package com.mrz.test;
import java.util.Arrays;
/*
* 2022-04-07
* mrz
* 计数排序
* 在一个比较大的数字范围中,对其中一部分进行排序(例如在一万个员工中按年龄排序(年龄暂定0-60)适用于此场景)
*/
public class CountSort {
public static void main(String[] args) {
//普通版本结果
int[] arr = {0,3,2,6,4,5,3,8,9,2,4,1}; //取值范围0-9
int[] result = sort(arr);
System.out.println(Arrays.toString(result));
//优化版本结果
int[] acc = {109,103,101,100,106,107,108,103,103,106};
int[] value = excellent(acc);
System.out.println(Arrays.toString(value));
}
//普通版本
static int[] sort(int[] arr) {
// TODO Auto-generated method stub
int[] result = new int[arr.length]; //首先创建一个和原数组等大小的数组
int[] count = new int[10]; //桶数组用于将原始数组数字的出现频率保存起来
for(int i = 0; i < arr.length; i++) { //遍历原数组
count[arr[i]]++; //将原数组数字放在桶数组对应下标位置,没出现一次就自增一次,例如:原数组第一个数字是3,桶数组下标为3的位置自增,整个原数组3出现两次,所以桶数组下标为3的值是2
}
System.out.println(Arrays.toString(count)); //打印桶数组,打印结果就是原数组对应桶数组下标的值出现的次数
for(int i = 0,j = 0;i<count.length;i++) { //遍历桶数组进行排序
while(count[i]-- > 0) result[j++] = i; //桶数组的值就是原数组对应桶数组值的下标出现的次数,如果count[i]的值大于0,就把i值赋给新数组的j位置,然后i自减,j自增,直到结束
}
return result;
}
//优化版本(解决如果数组数字不是从0开始,比如从100-110之间的值,如果按照普通版本就会额外开辟0-100的空间,发生空间浪费的情况)
static int[] excellent(int[] acc) {
//找出数组acc中的最大值和最小值
int max = acc[0];
int min = acc[0];
for(int num : acc) {
max = Math.max(max, num);
min = Math.min(min, num);
}
//初始化计数数组count
//长度为最大值减最小值加1(因为最大值减去最小值得到的是除最小值以为的数,因为还要包含最小值本身,所以加1)
int[] count = new int[max-min+1];
//对计数数组各元素赋值
for(int num : acc) {
//acc中的元素要减去最小值,再做为新索引
count[num-min]++;
}
//创建结果数组
int[] result = new int[acc.length];
//创建结果数组的起始索引
int index = 0;
//遍历计数数组,将计数数组的索引填充到结果数组中
for(int i = 0;i < count.length;i++) {
while(count[i]>0) {
//再将减去的最小值补上(例如原值是101-109,但是我们开辟的新数组是0-10,那我们在对原数组的值进行排序时要把之前减掉
//的部分加上,所以新数组的0下标值是100+0,1位置的下标是100+1,2位置的下标是100+2,依次类推得到原址)
result[index++] = i+min;
count[i]--;
}
}
return result;
}
}
如果一个数组里面数字是1-103,但是只需要对100-103位置进行排序,就体现出来优化后方法的优势了,不需要和原数组开辟等大小的空间,只需要开辟100-103四个大小的数组就行了,图示如下:
如果不明白的话可以用少量数据先打断点执行一下看看运行状态,也可以私信或留言~~~