算法笔记九、基数排序代码

package com.hao.firstdemo.datastruct.sort;

import java.util.Arrays;

/**
 * @author haoxiansheng
 * @data 2020/5/9 8.40
 */
public class RadixSort {
    public static void main(String[] args) {
        int arr[] = {53, 3, 542, 748, 14, 214};
        System.out.println(Arrays.toString(radixSort(arr)));
    }

    /**
     * 基数排序
     *
     * @return
     */
    public static int[] radixSort(int[] arr) {
        // 得到最大的数
        int max = arr[0];
        for (int i = 0; i < arr.length; i++) {
            if (max < arr[i]) {
                max = arr[i];
            }
        }
        // 得到 最大数的位数 几位
        int maxLength = (max + "").length();

        // 定义一个二维数组标识10个桶
        // 为了防止在放的时候出现数据溢出,每一个二维数组包含的以为数组长度初始化为arr.length
        // 空间换时间
        int[][] bucket = new int[10][arr.length];

        // 为了记录每个桶记录的个数,定义一个一维数组存放桶中数组的元素
        int[] bucketElementCounts = new int[10];

        for (int i = 0,  n = 1; i < maxLength; i++, n *= 10) {
            // 针对每个元素的的对应位进行处理 第一次个位 第二次10位 以此类推
            for (int j = 0; j < arr.length; j++ ) {
                // 取出对应每个元素的对应位的值
                int digitOfElement = arr[j] /n % 10;
                bucket[digitOfElement][bucketElementCounts[digitOfElement]] = arr[j];
                bucketElementCounts[digitOfElement]++;
            }
            // 按照这个桶的顺序（一维数组的下标依次取出数据,放入原来的数组）
            int index = 0;
            for (int k = 0; k < bucketElementCounts.length; k++) {
                // 如果桶中有数据才放入到元数据
                if (bucketElementCounts[k] != 0) {
                    // 循环该桶即第k个桶（即第k个一维数组） 放入
                    for (int l = 0; l < bucketElementCounts[k]; l++) {
                        // 取出放入到arr
                        arr[index++] = bucket[k][l];
                    }
                }
                // 每轮后需要 bucketElementCounts[k] = 0
                bucketElementCounts[k] = 0;
            }

        }

        return arr;
    }
}