题目
难度级别:简单
给你两个有序整数数组 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]
提示:
-10^9 <= nums1[i], nums2[i] <= 10^9
nums1.length == m + n
nums2.length == n
解题思路
这道题通过双指针,倒叙遍历数组进行解决。len1是nums1当前的指针,len2时nums2当前的指针。这里需要注意边界情况就是,当len1<0后,但len还存在的情况。
例如:
nums1为[4,0,0,0], m为1,nums2为[3,5,6],n为3。这时候遍历完len2是大于0的,且len也是大于0的,我们在重组数组时,应该把nums1上前面多余的0去掉,在把nums2剩余的元素放到nums1前面去。
const merge = function(nums1, m, nums2, n) {
let len1 = m - 1
let len2 = n - 1
let len = m + n - 1
while (len2 >= 0 && len1 >= 0)
nums1[len--] = nums1[len1] < nums2[len2] ? nums2[len2--] : nums1[len1--]
if (len2 >= 0)
nums1.splice(0, len + 1, ...nums2.splice(0,len2+1))
nums1.splice(m+n)
};
优化一下
const merge = function(nums1, m, nums2, n) {
while(m > 0 && n > 0)
nums1[m+n-1] = nums1[m-1] >= nums2[n-1] ? nums1[--m] : nums2[--n]
if (n > 0)
nums1.splice(0 , n, ...nums2.splice(0, n))
};
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array