java实现桶排序

参考文章: https://www.jianshu.com/p/c4217456c224. 感谢.

 

import sort.SortUtil;
import sort.insert.InsertSort;

import java.util.Arrays;

public class BucketSort2 {
    // 每个桶内数据排序借助于其他排序算法
    private static final InsertSort insertSort = new InsertSort();
    private static final int BUCKET_SIZE = 5;

    public static void bucketSort2(int[] data){
        int min = SortUtil.getMin(data);
        int max = SortUtil.getMax(data);
        // 计算桶的数量, 
        int bucketCount = (max-min) / BUCKET_SIZE + 1;
        //  一个桶为一个数组, 多个桶实际上就是个二维数组
        // 注意此处二维数组中的第二维数组的初始大小为0
        int[][] buckets = new int[bucketCount][0];
        for(int i=0; i<data.length; i++){
            // 计算数据应该落在哪个桶里
            int index = (data[i] - min) / BUCKET_SIZE;
            buckets[index] = pushData(buckets[index], data[i]);
        }
        // 依次对每个桶内的数据进行排序, 并将排好序的数据看存入原数组中 
        sortData(buckets, data);
    }

    private static int[] pushData(int[] bucket, int data){
        // 每次有数据落入桶中,要动态创建数组进行扩容,此处影响效率,不知道有没有改进方法
        // 如果不使用数组使用链表保存数据,或许可以改进
        bucket = Arrays.copyOf(bucket, bucket.length+1);
        bucket[bucket.length-1] = data;
        return bucket;
    }

    private static void sortData(int[][] buckets, int[] data){
        int index = 0;
        for(int[] bucket : buckets){
            // lengtｈ　为０ 表示没有数据落入该桶中
            if(bucket.length == 0){
                continue;
            }
           insertSort.sort(bucket);
            for(int number : bucket){
                data[index++] = number;
            }
        }
    }

    public static void main(String[] args) {
        int[] data = {4, 9, 1, 6, 8, 16, 2, 3, 0, 14, 5, 10, 19,15, 19,20,24};
        BucketSort2 bucketSort = new BucketSort2();
        try {
            bucketSort.bucketSort2(data);
            System.out.println("result: "+Arrays.toString(data));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}

 

 

public class InsertSort {
    
    public static int[] sort(int[] data) {
        int length = data.length;
        for(int i=1; i<length; i++){
            if(data[i]<data[i-1]){
                int temp = data[i];
                int j=i-1;
                for(; j>=0&&data[j]>temp; j--){
                    data[j+1] = data[j];
                }
                data[j+1] = temp;
            }
        }
        return data;
    }
}