给你两个有序整数数组 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除了m个元素外,还有多余空间。第一种思路是从前往后遍历两个数组,哪个小就留哪个,但是这样做要在nums1原有元素中间插值,麻烦。
所以最好是用两个指针,从后往前遍历两个数组。需要两个指针变量i,j。然后在nums1里从后往前填内容,比如两个数组一共六个元素,那就从nums1的i=5索引处往前填充元素。用i,j指针,哪个大,就把哪个放在要填的位置上。然后都前移,填充下一个位置。
var merge = function(nums1, m, nums2, n) {
let tar=m+n-1,i=m-1,j=n-1
while(i>=0 && j>=0){
if(nums1[i]<nums2[j]){
nums1[tar]=nums2[j]
j--
}else{
nums1[tar]=nums1[i]
i--
}
tar--
}
if(i===-1){
while(tar>=0){
nums1[tar]=nums2[j]
tar--
j--
}
}
};
首先循环遍历两个数组,只有当索引i和j都没有=-1,也就是都没出来的时候,才循环。只要有一个数组的元素全放完了,那就跳出循环。在循环体里比较,谁大就把谁放在tar索引处,然后指针前移,比较前一个数字。
跳出循环有两种可能:
-
j=-1,即数组2的元素全都放完了。那么此时数组1里剩下的不管几个数字,都肯定是最小的了,所以现在的nums1就不用动了,直接返回即可。
-
i=-1,即数组1的有效元素全都放完了,数组2里不知道剩下几个元素。因为数组2本来就是有序的,所以不管剩下几个,直接全都填到没填的位置即可。所以要单独判断i===-1的情况,循环只要tar>=0,就一直从数组2里取元素放到tar位置上。当tar<0,也就是说放到头了,那就跳出循环。