力扣088 合并两个有序数组

力扣088 合并两个有序数组

题干

给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。

说明：
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。 你可以假设 nums1 有足够的空间（空间大小大于或等于 m +
n）来保存 nums2 中的元素。

示例

输入：
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

输出：[1,2,2,3,5,6]

解析

1、取巧法

直接将两个数组合并，然后排序即可。简单粗暴。

但是取巧法存在问题，没有利用题干中数组有序这个条件。

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[:] = sorted(nums1[:m] + nums2[:])

2、双指针——需要额外空间

取巧法中提到，最好利用数组有序这个条件，而一看到数组有序，很自然得就想到双指针。

两个指针分别遍历nums1和nums2，依次比较大小，将较小的元素排列在一个新数组中。这样实现就利用了数组有序这个条件。

然而，本题要求对nums1做修改，而不是返回一个新数组，所以我们需要把nums1当成新数组来使用，那么nums1的内容就有可能在比较的过程中被修改。基于此考虑，可以先把nums1中有内容的部分拷贝出来，将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[:] = []
        i = 0
        j = 0
        while i < m and j < n:
            if nums1_copy[i] <= nums2[j]:
                nums1.append(nums1_copy[i])
                i += 1
            else:
                nums1.append(nums2[j])
                j += 1
        if i < m :
            nums1[i+j:] = nums1_copy[i:]
        else:
            nums1[i+j:] = nums2[j:]

需要注意的是，双指针会有一个指针先遍历完自己负责的数组，此时另一个数组中还有元素没有比较（或者说没有遍历到），这些没有被遍历到的元素都是大于此时nums1中现有元素的，因此要把没遍历的元素统一赋值到nums1的末尾。

3、双指针——不需要额外空间

使用双指针可以把时间复杂度控制在O（m+n），但是前一种双指针法用到了额外空间，如果不想使用额外空间，我们可以直接在nums1上进行操作。
但是nums1本身有内容，不能直接覆盖，但是观察nums1的后半部分，都是0（对于我们来说就是空闲的空间），我们可以使用这块空闲空间进行操作。

将双指针从后向前遍历，将较大元素放到nums1的最后，依次向前存放元素，这样就可以充分利用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.
        """
        i = m-1
        j = n-1
        k = m+n-1
        while i >= 0 and j >= 0:
            if nums1[i] > nums2[j]:
                nums1[k] = nums1[i]
                i -= 1
            else:
                nums1[k] = nums2[j]
                j -= 1
            k -= 1
        if i < j:
            nums1[:j+1] = nums2[:j+1] 

同样，会有一个指针先遍历完，此时需要将另一个数组中还未遍历的元素统一存放到nums1的最前端。