算法题 — 合并两个有序数组

合并两个有序数组

两个有数整数数组 nums1 和 nums2，将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。

初始化 nums1 和 nums2 的元素数量分别为 m 和 n。假设 nums1 的空间大小等于 m + n，这样它就能有足够的空间保存来自 nums2 的元素。

public static int[] merge(int[] nums1, int m, int[] nums2, int n) {
    // num2: 要复制的源数组; 0: 源数组要复制的起始位置(从0开始);
    // nums1: 要复制的目标数组 m: 目标数组的起始位置 n: 要复制的长度;
    System.arraycopy(nums2, 0, nums1, m, n);
    Arrays.sort(nums1);
    return nums1;
}

public static int[] merge1(int[] nums1, int m, int[] nums2, int n) {
    int[] nums1_copy = new int[m];
    System.arraycopy(nums1, 0, nums1_copy, 0, m);

    int p1 = 0; // 指向 nums1_copy
    int p2 = 0; // 指向 nums2
    int p = 0; // 指向 nums1

    while (p1 < m && p2 < n) {
        nums1[p++] = nums1_copy[p1] < nums2[p2] ? nums1_copy[p1++] : nums2[p2++];
    }

    if (p1 < m) {
        System.arraycopy(nums1_copy, p1, nums1, p1 + p2, m + n - p1 - p2);
    }
    if (p2 < n) {
        System.arraycopy(nums2, p2, nums1, p1 + p2, m + n - p1 - p2);
    }

    return nums1;
}

public static int[] merge3(int[] nums1, int m, int[] nums2, int n) {
    int p1 = m - 1;
    int p2 = n - 1;
    int p = m + n - 1;

    while (p1 >= 0 && p2 >= 0) {
        nums1[p--] = nums1[p1] < nums2[p2] ? nums2[p2--] : nums1[p1--];
    }
    System.out.println(Arrays.toString(nums1));
    System.out.println(Arrays.toString(nums2));

    System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
    return nums1;
}

int[] nums1 = new int[]{1, 3, 5, 7, 9, 0, 0, 0, 0};
int[] nums2 = new int[]{2, 4, 6, 8};
System.out.println(Arrays.toString(merge3(nums1, 5, nums2, 4)));