Leetcode解题之路(golang版)：88. 合并两个有序数组(Merge Sorted Array)

88. 合并两个有序数组

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

解法一：
按照归并排序的思想进行合并

$O(n + m)$

$O(n + m)$

//ind1,ind2,ind3分别记录下标
func merge(nums1 []int, m int, nums2 []int, n int)  {
    var ans = make([]int, m + n)
    copy(ans, nums1)
    ind1, ind2, ind3 := 0, 0, 0
    for ind1 < m || ind2 < n {
        if ind2 >= n || ind1 < m && ans[ind1] < nums2[ind2] {
            nums1[ind3] = ans[ind1]
            ind1++
        } else {
            nums1[ind3] = nums2[ind2]
            ind2++
        }
        ind3++
    }
    return
}

这里需要注意，ans为nums1的一个复制，但是不能使用等号复制，而是使用copy函数复制，因为等号复制只是地址复制，内存上ans和nums1还是相同的地方。

这个解法会使用额外的空间，但是本身nums1中已经有足够大的空间，所以可以改进一下这个算法，改进后如解法二。

解法二：
按照归并排序的思想进行合并，但是从后向前合并

$O(1)$

$O(n + m)$

//不适用额外空间，用nums1来记录，并且倒着合并
func merge(nums1 []int, m int, nums2 []int, n int)  {
    ind1, ind2, ind3 := m - 1, n - 1, n + m - 1
    for ind1 >= 0  || ind2 >= 0 {
        if ind2 < 0 || ind1 >= 0 && nums1[ind1] > nums2[ind2] {
            nums1[ind3] = nums1[ind1]
            ind1--
        } else {
            nums1[ind3] = nums2[ind2]
            ind2--
        }
        ind3--
    }
    return
}