真题描述:给你两个有序整数数组 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]
思路分析
双指针法。首先我们定义两个指针,各指向两个数组生效部分的尾部:每次只对指针所指的元素进行比较。取其中较大的元素,把它从 nums1 的末尾往前面填补:
如果提前遍历完的是 nums1 的有效部分,剩下的是 nums2。那么这时意味着 nums1 的头部空出来了,直接把 nums2 整个补到 nums1 前面去即可。
如果提前遍历完的是 nums2,剩下的是 nums1。由于容器本身就是 nums1,所以此时不必做任何额外的操作。
编码实现
/**
* @param {number[]} nums1
* @param {number} m
* @param {number[]} nums2
* @param {number} n
* @return {void} Do not return anything, modify nums1 in-place instead.
*/
var merge = function(nums1, m, nums2, n) {
let i = m - 1,j = n - 1, k = m + n - 1;
while(i>= 0 && j>=0){
if(nums1[i]>= nums2[j]){
nums1[k] = nums1[i];
k--;
i--;
}else{
nums1[k] = nums2[j];
k--;
j--;
}
}
while(j>=0){
nums1[k] = nums2[j]
k--;
j--;
}
return nums1
};
版权声明:本文为weixin_43939659原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。