88. 合并两个有序数组

2019/01/16 19:42 下午 posted in  leetcode

问题

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

说明:

  • 初始化 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+n-1\)处,而通过观察可以发现,\(m+n-1\)是必然大于等于\(m-1\)的。这样在倒序合并时,并不会覆盖掉\(nums1\)的原始数值的。这里的循环跳出条件是,\(nums1\)已经扫描完了,或者\(nums2\)已经扫描完了。但是需要注意的一点是,跳出后,如果是\(nums2\)扫描完了,那么可以直接返回;如果是\(nums1\)扫描完了,仍然需要把\(nums2\)的数据写回到\(nums1\)中去。

代码

java实现

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int i=m+n-1;
        while(m>0 && n>0) {
            if(nums1[m-1] >= nums2[n-1]){
                nums1[i--] = nums1[m-1];
                m--;
            } else {
                nums1[i--] = nums2[n-1];
                n--;
            }
        }
        while(n>0) {
            nums1[i--] = nums2[n-1];
            n--;
        }
    }
}

代码整体没有难度。注意哪个指针往前移就可以了。