题目
给你两个按 非递减顺序 排列的整数数组 和 ,另有两个整数 和 ,分别表示 和 中的元素数目。请你 合并 到 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 中。为了应对这种情况, 的初始长度为 ,其中前 个元素表示应合并的元素,后 个元素为 ,应忽略。 的长度为 。
解题
以下算法的时间复杂度是 ,空间复杂度是 ,因为我们直接在 数组上进行了操作。
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 -= 1else:nums1[p] = nums2[p2]p2 -= 1p -= 1