public class BucketSortDemo {
public static void bsa(int[] nums,int bucketSize) {
if (nums == null || nums.length == 0 || nums.length == 1){
return;
}
//获取最大值,最小值
int max = nums[0],min = nums[0];
for (int i=0;i<nums.length;i++){
max = Math.max(max,nums[i]);
min = Math.min(min,nums[i]);
}
//根据bucketSize获取桶的个数
int bucketNums = (max-min) / bucketSize + 1;
List<List<Integer>> bucketArray = new ArrayList<>(bucketNums);
for (int i=0;i<bucketNums;i++){
bucketArray.add(new ArrayList<Integer>());
}
//遍历数组,将元素存到其所属范围的桶内
for (int i=0;i<nums.length;i++){
bucketArray.get((nums[i]-min)/bucketSize).add(nums[i]);
}
//对每个桶的数据进行排序
for (int i=0;i<bucketNums;i++){
Collections.sort(bucketArray.get(i));
}
//对多个桶的数据从前到后遍历存到数组中
int index = 0;
for (int i=0;i<bucketArray.size();i++){
for (int j=0;j<bucketArray.get(i).size();j++){
nums[index++] = bucketArray.get(i).get(j);
}
}
}
public static void main(String[] args) {
int[] array = {3,4,6,1,7,5,2,34,98};
bsa(array,3);
for (int i=0;i<array.length;i++){
System.out.println(array[i]);
}
}
}
实现流程:
1.找到待排序数组的最大值(max)、最小值(min)
2.根据最大值、最小值的差值和指定的桶大小,新建多个桶
3.将待排序数组的数存到其相应范围的桶中
4.对每个桶的数据进行排序
4.对多个桶的数据从前到后遍历存到数组中
时间复杂度:O(n+k)
空间复杂度:O(n+k)