算法练习——合并数组 leetcode.88 python

题目描述：

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

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

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

方法一：将nums2和nums1合并 直接调用python的sort方法 这也是最简单的方法

def merge(nums1,nums2,m,n):
    i = m
    j = 0
    while i < m+n:
        nums1[i] = nums2[j]
        i = i + 1
        j = j + 1
    nums1.sort()

击败了26%的用户。

方法二：自写排序方法

def merge(nums1,nums2,m,n):
    temp = []
    i = 0
    j = 0
    while i < m or j < n:
        if i >= m: # nums1已经排完了（到0了）
            temp.append(nums2[j])
            j = j + 1
        elif j >= n: # nums2已经排完了
            temp.append(nums1[i])
            i = i + 1
        elif nums1[i] < nums2[j]:
            temp.append(nums1[i])
            i = i + 1
        elif nums1[i] >= nums2[j]:
            temp.append(nums2[j])
            j = j + 1
        k = 0
        # 题目要求返回nums1
        while k < m+n:
            nums1[k] = temp[k]
            k = k + 1

击败了99%的用户。

（令我疑惑的是，如果最后一步 直接 nums1 = temp 在pycharm里运行是正确的，但在了力扣里错误。）

这个写法的缺陷是，过程中有一个临时数组temp占用内存，如下的方法可以做进一步改进。

方法三：思路仍然是两个指针，如果没有临时数组，那应该向哪里存储数据呢——我们想到，nums1后留有[0,0,0]，我们可以从后向前降序排列。

def merge(nums1,nums2,m,n):
    i = m - 1
    j = n - 1
    k = m + n - 1
    while k >= 0: # 如果仅仅是k > 0 这种情况过不了 [0]、[1]合并
        if i < 0:
            nums1[k] = nums2[j]
            j = j - 1
        elif j < 0:
            nums1[k] = nums1[i]
            i = i - 1
        elif nums1[i] < nums2[j]:
            nums1[k] = nums2[j]
            j = j - 1
        elif nums1[i] >= nums2[j]:
            nums1[k] = nums1[i]
            i = i - 1
        k = k - 1
    return nums1

在空间复杂度上有一定改观。