import insertionSort from './insertion-sort.mjs';

/**
 * 使用lsd基数排序
 * @param {number[]} arr - 要基数排序的数组
 */
export function radixSortByLsd(arr) {
  let max = Math.abs(Math.max(...arr)); // 拿取数组中最大的数，用于计算长度
  let buckets = Array.from({ length: 10 }, () => []);
  let curDit = 1; // 当前比较的是哪个数位 1表示个位
  /* 处理负数，思路： 把所有的负数和正数都分开，负数使用绝对值处理之后以同样的方法处理，处理完之后取相反数再
    倒序添加到原数组的前面
  */
  let negativeArr = [];
  const tempArr = [];
  for (let i = 0; i < arr.length; i++) {
    const temp = arr[i];
    if (temp < 0) {
      negativeArr.push(temp);
    } else {
      tempArr.push(temp);
    }
  }
  arr = tempArr;
  if (negativeArr.length) {
    negativeArr = radixSortByLsd(negativeArr.map((_) => -_)).map((_) => -_);
  }
  while (Math.floor(max)) {
    for (let i = 0; i < arr.length; i++) {
      buckets[getNumByGidit(arr[i], curDit)].push(arr[i]); // 根据数位的大小放入对应的数位桶中
    }
    // 对buckets中的每个桶的元素做好排序，这里我们调用之前的插入排序
    for (let i = 0; i < buckets.length; i++) {
      insertionSort(buckets[i]);
    }
    // 个位比较完成之后比较十位，在此之前要把个位有序的数组给原数组,并且把桶子清空
    arr = buckets.flat();
    buckets = Array.from({ length: 10 }, () => []);
    max /= 10; // 处理下一数位
    curDit++;
  }
  return negativeArr.length ? negativeArr.reverse().concat(arr) : arr;
}

/**
 * @description 获取某位的数字,从1开始，1代表是个位，比如（100，3）返回的是1，3代表千位
 * @param {number} num - 传入的数值
 * @param {number} dit - 要获取的位数
 */
function getNumByGidit(num, dit) {
  if (!num) return 0;
  let res = 0;
  while (dit--) {
    res = num % 10;
    num = (num - res) / 10;
  }
  return res;
}

/**
 * @description 使用msd的基数排序
 * @param {number[]} arr - 需要排序的数组
 */
export function radixSortByMsd(arr) {
  // 获取最大值，并通过最大值对小的数补零
  const maxVal = Math.max(...arr);
  const n = arr.length; // 数组的个数
  const radix = Math.floor(Math.log10(maxVal)); // 对应的位数，比如100就是2代表百分位
  // 从高位到地位开始比较，分别放入对应的桶中，再便利，使之相对有序
  const buckets = Array.from({ length: 10 }, () => []);
  for (let i = 0; i < n; i++) {
    buckets[getNumByGidit(arr[i], radix + 1)].push(arr[i]);
  }
  // 对每一个桶中的内容进行msd基数排序，如果零号桶还有的话
  if (buckets[0].length) {
    for (let i = 0; i < buckets.length; i++) {
      buckets[i] = radixSortByMsd(buckets[i]);
    }
  }
  return buckets.flat();
}