88. 合并两个有序数组
两个有序整数数组 nums1 和 nums2,请将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
//从后往前比较元素,并从后往前填充num1切片
func merge(nums1 []int, m int, nums2 []int, n int) {
//从后往前遍历
for i := n + m - 1; i >= 0; i-- {
//当两个切片都没有遍历完时
if m-1 >= 0 && n-1 >= 0 {
//当num1的元素>=num2中的元素时,将该元素从后往前填入num1切片
if nums1[m-1] >= nums2[n-1] {
nums1[i] = nums1[m-1]
m--
} else {
//当num2的元素>num1中的元素时,将该元素从后往前填入num1切片
nums1[i] = nums2[n-1]
n--
}
} else {
//当其中一个切片遍历结束时,存在两种情况
//1.当num2先结束,由于num1和num2的元素都是从小到大排列的,故num1剩下的元素可以原封不动
//2.当num1先结束,num2元素还有剩余,则将num2剩下的元素继续存入num1切片中
if m-1 < 0 && n-1 >= 0 {
nums1[i] = nums2[n-1]
n--
}
}
}
}