《leetcode88. 合并两个有序数组》

236 阅读2分钟

给你两个有序整数数组 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索引处,然后指针前移,比较前一个数字。

跳出循环有两种可能:

  1. j=-1,即数组2的元素全都放完了。那么此时数组1里剩下的不管几个数字,都肯定是最小的了,所以现在的nums1就不用动了,直接返回即可。

  2. i=-1,即数组1的有效元素全都放完了,数组2里不知道剩下几个元素。因为数组2本来就是有序的,所以不管剩下几个,直接全都填到没填的位置即可。所以要单独判断i===-1的情况,循环只要tar>=0,就一直从数组2里取元素放到tar位置上。当tar<0,也就是说放到头了,那就跳出循环。