Python 基础算法——合并两个有序数组

10. 合并两个有序数组

示例：

输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出：[1,2,2,3,5,6]
解释：需要合并 [1,2,3] 和 [2,5,6] 。
合并结果是 [1,2,2,3,5,6] 

提示：

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

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

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。

为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

常用的两种算法分别是：排序法和逆向双指针法。

排序法：

排序法的思路是，先把 nums2 中的元素复制到 nums1 的尾部，然后对 nums1 进行排序，最后返回排序后的 nums1。

时间复杂度是 O(nlogn) 的原因是，这个方法需要对 nums1 进行排序，排序的时间复杂度一般是 O(nlogn)。

空间复杂度是 O(1) 的原因是，这个方法不需要额外的空间来存储数组中的元素，只需要在原数组上进行操作。

# 排序法    时间复杂度:O(nlogn)     空间复杂度: O(1)
def merge1(nums1, m, nums2, n):
    # 将 nums2 中的元素复制到 nums1 的尾部
    for i in range(n):
        nums1[m + i] = nums2[i]
    # 排序
    nums1.sort()

逆向双指针法：

逆向双指针法的思路是，从数组的尾部开始合并，而不是从头部开始。这样可以避免移动数组中的元素，节省时间和空间。

时间复杂度是 O(m + n) 的原因是，这个方法最多需要遍历两个数组中的所有元素，对于每个元素，只需要进行常数次的比较和赋值操作。

空间复杂度是 O(1) 的原因是，这个方法不需要额外的空间来存储数组中的元素，只需要维护三个指针，分别指向 nums1、nums2 的尾部和合并后的尾部。

def merge2(nums1, m, nums2, n):    # 时间复杂度: O(m + n)      空间复杂度: O(1)。
    # 先定义三个指针，分别指向 nums1， nums2 的尾部和合并后的尾部
    p1 = m - 1
    p2 = n - 1
    p = m + n - 1
    # 循环直到遍历完其中一个数组
    while p1 >= 0 and p2 >= 0:
        # 比较两个指针所指元素的大小，将较大的元素放到合并后的尾部，并移动相应的指针
        if nums1[p1] > nums2[p2]:
            nums1[p] = nums1[p1]
            p1 -= 1
        else:
            nums1[p] = nums2[p2]
            p2 -= 1
        p -= 1
    # 如果 nums2 还有剩余元素，将它们复制到 nums1 的前部
    while p2 >= 0:
        nums1[p] = nums2[p2]
        p -= 1
        p2 -= 1

最后一个循环的作用：

这段代码的目的是为了处理一种特殊情况，就是当 nums2 中还有剩余元素，而 nums1 中已经没有可以合并的元素时，需要将 nums2 中的剩余元素复制到 nums1 的前部。😊

这种情况可能发生在 nums2 中的元素都比 nums1 中的元素小的时候。例如，假设 nums1 = [4, 5, 6, 0, 0, 0]，m = 3，nums2 = [1, 2, 3]，n = 3。按照逆向双指针法的思路，我们从数组的尾部开始合并，得到以下过程：

• p1 指向 nums1 的第三个元素，即 6；p2 指向 nums2 的第三个元素，即 3；p 指向合并后的尾部，即 nums1 的第六个元素。
• 比较 p1 和 p2 所指元素的大小，发现 p1 大于 p2，所以将 p1 所指元素复制到 p 所指位置，并将 p 和 p1 都左移一位。此时 nums1 = [4, 5, 6, 0, 0, 6]。
• 比较 p1 和 p2 所指元素的大小，发现 p1 大于 p2，所以将 p1 所指元素复制到 p 所指位置，并将 p 和 p1 都左移一位。此时 nums1 = [4, 5, 6, 0, 5, 6]。
• 比较 p1 和 p2 所指元素的大小，发现 p1 大于 p2，所以将 p1 所指元素复制到 p 所指位置，并将 p 和 p1 都左移一位。此时 nums1 = [4, 5, 6, 4, 5, 6]。

到这里，我们发现 nums1 中还有一个剩余元素，即第一个元素 4，而 nums2 中已经没有可以合并的元素了（p2 已经小于 0）。如果我们不执行这段代码，那么我们就会丢失 nums2 中的所有元素，导致合并后的数组不完整。

题目来源：力扣