题目
给你两个按 非递减顺序 排列的整数数组
和
,另有两个整数
和
,分别表示
和
中的元素数目。请你 合并
到
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组
中。为了应对这种情况,
的初始长度为
,其中前
个元素表示应合并的元素,后
个元素为
,应忽略。
的长度为
。
解题
以下算法的时间复杂度是
,空间复杂度是
,因为我们直接在
数组上进行了操作。
def merge(nums1, m, nums2, n):
# 设置两个指针,分别指向 nums1 和 nums2 的最后一个有效元素
p1, p2 = m - 1, n - 1
# 设置指针指向 nums1 的最后一个位置
p = m + n - 1
# 从后往前遍历,直到 nums2 处理完毕
while p2 >= 0:
# 比较 nums1 和 nums2 当前指针指向的值,选择较大者放入 nums1[p] 位置
if p1 >= 0 and nums1[p1] > nums2[p2]:
nums1[p] = nums1[p1]
p1 -= 1
else:
nums1[p] = nums2[p2]
p2 -= 1
p -= 1