题干

给你两个按 非递减顺序 排列的整数数组 nums1 **和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 **到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意： 最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

题解

比较简单的想法是完全可以开一个新的数组，合并完了之后再把这个新的数组赋值给nums1，但是这样空间复杂度就是O(m+n)了，并不是最优解。题目要求结果保存在nums1中，是在暗示可以原地操作，使用O(1)的空间复杂度解决这个问题。nums1的长度为m+n，且后n个数是0，因此，可以反向遍历nums1和nums2，取较大的数，从nums1的尾部开始填充结果。

import "fmt"

func merge(nums1 []int, m int, nums2 []int, n int) {
	i := m - 1
	j := n - 1
	k := m + n - 1
	for k >= 0 || i >= 0 || j >= 0 {
            // 如果nums1取完了，就取nums2; 如果nums2取完了，就取nums1;否则就取两个列表中较大的值
		if i == -1 {
			nums1[k] = nums2[j]
			j--
		} else if j == -1 {
			nums1[k] = nums1[i]
			i--
		} else if nums1[i] >= nums2[j] {
			nums1[k] = nums1[i]
			i--
		} else {
			nums1[k] = nums2[j]
			j--
		}
		k--
	}
}

func main() {
	nums1 := []int{1, 2, 3, 0, 0, 0}
	m := 3
	nums2 := []int{2, 5, 6}
	n := 3
	merge(nums1, m, nums2, n)
	fmt.Println(nums1)

	nums1 = []int{1}
	m = 1
	nums2 = []int{}
	n = 0
	merge(nums1, m, nums2, n)
	fmt.Println(nums1)

	nums1 = []int{0}
	m = 0
	nums2 = []int{1}
	n = 1
	merge(nums1, m, nums2, n)
	fmt.Println(nums1)

	nums1 = []int{4, 5, 6, 0, 0, 0}
	m = 3
	nums2 = []int{1, 2, 3}
	n = 3
	merge(nums1, m, nums2, n)
	fmt.Println(nums1)

	nums1 = []int{4, 0, 0, 0, 0, 0}
	m = 1
	nums2 = []int{1, 2, 3, 5, 6}
	n = 5
	merge(nums1, m, nums2, n)
	fmt.Println(nums1)
}