算法筆記九、基數排序代碼

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;
    }
}