给你两个按非递减顺序排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2中的元素数目。请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序
排列。
ps: 最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
实现如下:
方法1. 直接合并后排序
时间复杂度为O((m+n)*log(m+n))
function mergeOfSort(nums1, m, nums2, n) {
nums1.splice(m, nums1.length - m, ...nums2);
nums1.sort((a, b)=> a - b);
}
let nums1 = [1,2,3,0,0,0], m = 3;
let nums2 = [2,5,6], n = 3;
// mergeOfSort(nums1, m, nums2, n)
// mergeOfDoubleKey(nums1, m, nums2, n)
mergeOfLastDoubleKey(nums1, m, nums2, n)
console.log(nums1);
方法2. 双指针
时间复杂度为O(m+n) 指针移动单调递增,最多移动m+n次。空间复杂度:O(m+n)。需要建立长度为m+n的中间数组sorted。
function mergeOfDoubleKey(nums1, m, nums2, n) {
let p1 = 0, p2 = 0; // 分别的nums1和nums2的指针
const sorted = new Array(m + n).fill(0);
let cur;
while (p1 < m || p2 < n) {
if (p1 === m) {
cur = nums2[p2++];
}else if (p2 === n) {
cur = nums1[p1++]
} else if (nums1[p1] < nums2[p2]) {
cur = nums1[p1++]
} else {
cur = nums2[p2++]
}
sorted[p1+p2-1] = cur;
}
for (let i = 0; i !== m+n ; i++) {
nums1[i] = sorted[i]
}
}
方法3. 逆向双指针(最优)
时间复杂度:O(m+n) 空间复杂度:O(1)直接对数组nums1原地修改,不需要额外空间。
function mergeOfLastDoubleKey(nums1, m, nums2, n) {
let p1 = m - 1, p2 = n - 1;
let tail = m + n - 1; // 表示最长数组的当前位置
let cur;
while (p1 >= 0 || p2 >= 0) {
if (p1 === -1) {
cur = nums2[p2--];
} else if (p2 === -1) {
cur = nums1[p1--];
} else if (nums1[p1] > nums2[p2]) {
cur = nums1[p1--]
} else {
cur = nums2[p2--]
}
nums1[tail--] = cur; // 从最后一位开始赋值
}
}