6 两个数组的差_算法题解之合并两个有序数组

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

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间（空间大小大于或等于 m + n）来保存 nums2 中的元素。

示例：

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]

题解分析

方法一：直接合并

将 num2的元素直接添加到 nums1中后，再进行排序，此方法简单，但是时间复杂度比较差

class Solution {
  public void merge(int[] nums1, int m, int[] nums2, int n) {
    //将num2 copy 至 nums1
    System.arraycopy(nums2, 0, nums1, m, n);
    //排序nums1
    Arrays.sort(nums1);
  }
}

执行结果

方法二：双指针法

此题目其实是可以使用双指针的方式

指定两个指针p1, p2分别代表 m-1及n-1即数组 nums1及 nums2的最后下标，p 则是 m+n-1。

当 nums1[p1] < nums2[p2]时，将 nums2[p2]的值放入nums[p]中， 同时 p、p2向左移，即 p--、p2--;

当 nums1[p1] > nums2[p2]时，则将 num1[p1]的值放入 nums[p]中，同时p、p1向左移，即 p--、p1--;

最后再将nums2多余的数据复制至 num1的开头即可（因为是有序的且都比nums1的值小，直接复制过来就可以）

第一步：

第二步：

第三步：

后面依此类推，即可

class Solution {
  public void merge(int[] nums1, int m, int[] nums2, int n) {
    // 定义两个指针变量
    int p1 = m - 1;
    int p2 = n - 1;
    // 定义 nums1存数的变量
    int p = m + n - 1;

    // 循环开始
    while ((p1 >= 0) && (p2 >= 0))
      // 比较两个数的大小
      // 如果nums1[p1]比nums[p2]小，则p2赋值并减1且p减1
      // 如果nums1[p1]比nums[p2]大，则p1赋值并减1且p减1
      nums1[p--] = (nums1[p1] < nums2[p2]) ? nums2[p2--] : nums1[p1--];

    // 将 nums2多余的元素复制到 nums1开头
    System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
  }
}

执行结果