Leetcode每日一题-合并两个有序数组

题目描述

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

说明:

• 初始化 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]

解答

方法一：先合并再排序

var merge = function(nums1, m, nums2, n) {
  for (var i = 0; i < nums2.length; i++) nums1[m + i] = nums2[i]
  nums1.sort((a, b) => a - b)
}

或者用push插入数组

var merge = function(nums1, m, nums2, n) {
  nums1.splice(m)
  for (let i = 0; i < n; i++) {
    nums1.push(nums2[i])
  }
  nums1.sort((a, b) => a - b)
};

方法二：双指针 + 从后向前

思路：

1. 两个分别指向两个数组尾部的指针从后向前
2. 比较两指针位置的值：大的一定是结果数组的最大值，填充到 nums1的末尾
3. 遍历完后：当 n > 0 时，说明 nums2 中还有剩余没有比较的数字，将其插入替换 nums1 数组前面n个数字即可

var merge = function(nums1, m, nums2, n) {
    let count = m + n;
    while(m > 0 && n > 0){
        nums1[--count] = nums1[m-1] < nums2[n-1] ? nums2[--n] : nums1[--m];
    }
    if(n > 0){
        nums1.splice(0,n,...nums2.slice(0,n));
    }
};

参考：leetcode题解