LeetCode 88.合并两个有序数组

一、题目详情

给你两个有序整数数组 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]

二、思路

1.之前我们做过一道类似的题目LeetCode 21.合并两个有序链表，不过这一次是换成了两个有序数组。
按照那一题的思路，我们可以额外定义一个数组 temp[]，然后依次比较 nums1[] 和 nums2[] 中的数据，将较小的数据放入 temp[] 之中。

while(i<m&&j<n) {
    if(nums1[i]<nums2[j]) {
        temp[k++]=nums1[i++];
    }else {
        temp[k++]=nums2[j++];
    }
}

直到其中一个数组为空为止，再将另一个数组剩下的数据拷贝进 temp[] 中。

if(i==m) {//nums1空了
    while(j<n) {
        temp[k++]=nums2[j++];
    }
}else{//nums2空了
    while(i<m) {
        temp[k++]=nums1[i++];
    }
}

最后将 temp[] 中的数据全部赋值给 nums1[] 就好。

for(i=0;i<m+n;i++) { 
    nums1[i]=temp[i];
}

2.记得之前我们还有一道题，LeetCode 73.矩阵置零，将O(mn)的额外空间，优化成O(m+n)，后续又优化为O(1)。

这里我们也有一个改进方案，不额外定义 temp[] 数组来存储已经排序的数据，而是直接使用 nums1[] 来存储。
但是如果从前向后遍历，会覆盖 nums1[] 中原先的元素，所以我们从后向前遍历，每次取较大的数据放入 nums1[] 数组的最后面。

三、代码

public void merge(int[] nums1, int m, int[] nums2, int n) {
    int i=m-1,j=n-1,k=m+n-1;
    //两个数组有一个不为空，都继续比较
    while(i>=0||j>=0) {
        if(i<0) {
            //nums1原先数据已经比较完了
            nums1[k--]=nums2[j--];
        }else if(j<0) {
            //nums2原先数据已经比较完了
            nums1[k--]=nums1[i--];
        }else if(nums1[i]>nums2[j]) {
            //取两个数的较大值
            nums1[k--]=nums1[i--];
        }else {
            nums1[k--]=nums2[j--];
        }
    }
}