[leetcode]88.Merge Sorted Array

Posted 2021-01-09 shinjia

题目

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.
Example:

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

Output: [1,2,2,3,5,6]

解法一

思路

最先想到的是把num2直接复制到nums1后面，然后进行个排序即可。时间复杂度取决于排序算法。

代码

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int j = 0;
        for(int i = m; i < m+n; i++) {
            nums1[i] = nums2[j++];
        }
        Arrays.sort(nums1);
    }
}

解法二

思路

这种方法直接创建一个新的m+n的数组，然后比较num1和nums2的当前元素，将较小的值插入新数组，依次循环即可。但是这种方法违反了题目的本意，因为题目并不想让创建新数组，所以此种方法不写代码了，了解即可。

解法三

思路

这种方法比较精妙，由于合并后的数组必然是m+n的，所以我们可以从后往前开始赋值，先将nums1和nums2的较大值放到m+n-1的位置上，然后依次向前推。如果循环结束，当n不为0但m为0时，num1剩下的元素就在相应的位置，不用处理。当m不为0但n为0，那就把nums2剩下的数依次复制过去。

代码

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