1、描述
给定两个有序数组 nums1 和 nums2 ,将 nums2 合并到 nums1 中,使得 nums1 成为一个有序数组。
说明:
1)初始化nums1 和 nums2的元素数量分别为 m 和 n
2)你可以假设 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, 0, 0, 3, 3, 3, 0, 0, 0], m = 6
nums2 = [1, 2, 2], n = 3
输出:[-1, 0, 0, 1, 2, 2, 3, 3, 3]
2、算法
解法一:合并后排序
思想:将nums2的n个数合并到nums1的m~n-1下标位置,再对nums1排序
时间复杂度:O(n)
func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
/*
合并后排序
*/
nums1 = [Int](nums1[0..<m])
for num in nums2 {
nums1.append(num)
}
nums1 = nums1.sorted()
print(nums1)
}
解法二:从前往后排序
思想:比较两个数组的第一个元素,然后加入新的数组,移除旧数组中的,最后判断是否有剩下的,若有全部加入新数组,最后将新数组赋值给nums1
时间复杂度:O(n)
func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
/*
从前往后排序
*/
if m==0 {
nums1 = nums2
}
var num = [Int]()
var newNum1 = [Int](nums1[0..<m])
var nums2 : [Int] = nums2
while !newNum1.isEmpty && !nums2.isEmpty {
if newNum1.first! <= nums2.first!{
num.append(newNum1.first!)
newNum1.removeFirst()
}else{
num.append(nums2.first!)
nums2.removeFirst()
}
if newNum1.isEmpty {
num.append(contentsOf: nums2)
nums1 = num
}
if nums2.isEmpty{
num.append(contentsOf: newNum1)
nums1 = num
}
}
print(nums1)
}
解法三:从后往前排序
思想:利用双指针,将最大的数加入nums1,依次循环
时间复杂度
func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
//双指针 从后往前
var p1 = m-1
var p2 = n-1
var p = m+n-1
while p1>=0 && p2>=0 {
if nums1[p1] <= nums2[p2]{
nums1[p] = nums2[p2]
p2 -= 1
p -= 1
}else{
nums1[p] = nums1[p1]
p1 -= 1
p -= 1
}
}
while p2 >= 0 {
nums1[p] = nums2[p2]
p -= 1
p2 -= 1
}
}