Leetcode每日一题2021.2.28第88题：合并两个有序数组

题目描述

示例

方法一：双指针/从前往后

将nums1的前m个元素复制一份，放在一个新列表nums1_copy中，再清空nums1。设置指针p1和p2，p1指向nums1_copy的开头，p2指向nums2的开头，在每一步将最小值放出输出数组nums1中。

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        nums1_copy = nums1[:m]
        # 注意：题目要求在nums1上原地修改
        nums1[:] = []

        p1 = p2 = 0
        while p1 < m and p2 < n:
            if nums1_copy[p1] < nums2[p2]:
                nums1.append(nums1_copy[p1])
                p1 += 1
            else:
                nums1.append(nums2[p2])
                p2 += 1

        # 如果还有数没加进去的话
        if p1 < m:
            nums1[p1+p2:] = nums1_copy[p1:]
        if p2 < n:
            nums1[p1+p2:] = nums2[p2:]

方法二：双指针/从后向前

方法一已经取得了最优的时间复杂度O(n+m)，但需要使用额外空间。
如果我们从结尾开始改写num1的值的话，因为这里没有信息，所以不需要额外的空间。
这里用指针p用于追踪添加元素的位置。

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        p1 = m - 1
        p2 = n - 1
        p = m + n - 1
        while p1 >= 0 and p2 >= 0:
            if nums1[p1] < nums2[p2]:
                nums1[p] = nums2[p2]
                p2 -= 1
            else:
                nums1[p] = nums1[p1]
                p1 -= 1

            p -= 1

        # 如果p1先到头了，说明nums1前面的数都比nums2剩下的大，且已经全放到nums1后面了，可以把nums2剩下的全放在 
        # nums1前面了。 
        # 如果p2先到头了，那下面的操作等于进行了一个寂寞。
        nums1[:p2+1] = nums2[:p2+1]