排序算法-基数排序和计数排序

题目:基数排序

public static int[] baseSort(int[] arr){
        // PC
        if(arr == null || arr.length<=0)
            return null;
        int[] help = new int[arr.length];
        // 求最大值 得到最大位数
        int max = getMax(arr);
        // 得到最高位
        int base = getBase(max);
        int[] offset = new int[10];
        for (int i = 1; i <= base; i++) {
            // 记录各个base数量 更新offset 该数组用于记录arr元素最终放在数组哪个位置
            for (int j : arr) {
                int index = getNum(j, i);
                offset[index] ++;
            }
            for (int j = 1; j < offset.length; j++) {
                offset[j] += offset[j-1];
            }
            for (int length = arr.length-1; length >= 0; length--) {
                help[--offset[getNum(arr[length],i)]] = arr[length];
            }
            offset = new int[10];
            for (int j = 0; j < help.length; j++) {
                arr[j] = help[j];
            }
        }
        return arr;
    }
    public static int getNum(int num,int base){
        num /= Math.pow(10,base-1);
        return num%10;
    }
    public static int getMax(int[] arr){
        int max = Integer.MIN_VALUE;
        for (int i : arr) {
            max = Math.max(i,max);
        }
        return max;
    }
    public static int getBase(int n){
        int count = 0;
        while (n>0){
            count++;
            n /= 10;
        }
        return count;
    }

题目:计数排序

思路:每个元素作为数组index,对应位置值累加,最后遍历count数组

    public static int[] countSort(int[] arr){
        if(arr == null || arr.length <=0)
            return null;
        int min = Integer.MAX_VALUE;
        int max = Integer.MIN_VALUE;
        for (int i : arr) {
            min = Math.min(min,i);
            max = Math.max(max,i);
        }
        int len = (max-min)+1;
        int[] count = new int[len];
        // base 就是min
        for (int i : arr) {
            count[i-min]++;
        }
        int[] ans = new int[arr.length];
        int index = 0;
        for (int i = 0; i < count.length; i++) {
            while (count[i]>0){
                // 多个相同的放在一起
                ans[index++] = i+min;
                count[i]--;
            }
        }
        return ans;
    }

你可能感兴趣的:(算法,排序算法,算法,数据结构)