在归并排序中对小数组采用插入排序

主体还是归并排序,使用插入排序粗化原子任务的叶子节点

public static void main(String[] args) {
    // 构造数据
    Random random =  new Random(System.currentTimeMillis());
    int[] arr = new int[1000000];
    for (int i = 0; i < arr.length; i++) {
        arr[i] = random.nextInt();
    }
    // 测试数组
    int[] arr2 = new int[arr.length];
    // 步长为 5,观察寻找最优解 k 值, 大部分情况下在100左右出现
    for (int i = 1; i <= 500; i += 5) {
        System.arraycopy(arr, 0, arr2, 0, arr.length);
        long start = System.currentTimeMillis();
        mergeWithInsert(arr2, 0, arr2.length - 1, i);
        long end  = System.currentTimeMillis();
        System.out.println("k:["  + i + "] -> " + (end - start));

    }
}


/**
 *
 * @param arr 数组
 * @param p 起始节点下标
 * @param r 结束节点下标
 * @param k 当 r+1-q <= k 时,不再分解问题,直接使用insert解决问题
 */
public static void mergeWithInsert(int[] arr, int p, int r, int k) {
    if (p < r) {
        // 步长小于k, 则直接采用插入排序, 制造粗叶子节点
        if ((r + 1 - p) <= k) {
            insertionSort(arr, p, r);
        } else {
            // 计算q,分组
            int q = (p + r) / 2;
            // 排序左边
            mergeWithInsert(arr, p, q, k);
            // 排序右边
            mergeWithInsert(arr, q + 1, r,  k);
            // 合并左右结果
            merge(arr, p, q, r);
        }
    }
}

/**
 * arr 原始数组, 不使用哨兵牌,判断条件改为i,j两个索引必须同时小于各自数组的长度
 * @param arr 原始数组
 * @param p 当前Left分组的起始点下标
 * @param q 当前中间节点下标
 * @param r 当前Right分组的结束点下标
 */
public static void merge(int[] arr, int p, int q, int r) {
    // 计算两个分组的长度
    int n1 = q - p + 1;
    int n2 = r - q;
    // 复制原数组对应端的数据到两个新的分组
    int[] lArr = new int[n1];
    int[] rArr = new int[n2];
    System.arraycopy(arr, p, lArr, 0, n1);
    System.arraycopy(arr, q + 1, rArr, 0, n2);
    // 开始排序
    int i = 0, j = 0, k;
    // 遍历当前排序区间
    for (k = p; i < n1 && j < n2; k++) {
        // 左侧拿牌
        if (lArr[i] <= rArr[j]) {
            arr[k] = lArr[i];
            i++;
        } else { // 右侧拿牌
            arr[k] = rArr[j];
            j++;
        }
    }
    // 拿未拿完的堆的牌到排序堆顶
    if (i == n1) {
        System.arraycopy(rArr, j,arr, k, n2  - j);
    } else {
        System.arraycopy(lArr, i,arr, k, n1  - i);
    }
    // 打印每一次merge的结果
    // System.out.println(Arrays.toString(arr));
}


private static void insertionSort(int[] arr, int p, int r) {
    int i;
    for (int j = p + 1; j <= r; j++) {
        i = j - 1;
        int key = arr[j];
        while (i >= p && arr[i] > key) {
            arr[i + 1] = arr[i];
            i = i - 1;
        }
        arr[i + 1] = key;
    }
}

你可能感兴趣的:(在归并排序中对小数组采用插入排序)