合并两个有序数组

问题来源：力扣算法面试汇总

问题描述：给定两个有序整数数组 nums1 和 nums2，将 nums2 合并到 nums1 中，使得 num1 成为一个有序数组。

说明:
初始化 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]

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

1. 暴力算法

思路：一个较为简单的思路是，将nums1[:m]和nums2拼接起来，记为nums，然后对nums进行排序。实际上，对于任何一个列表，有许多方法进行排序，诸如选择排序、插入排序、归并排序等等。

由于方法较多，我们并不写出详细代码。

2. $O(n+m)$+额外空间

思路：第一类算法实际上没有利用nums1和nums2都是有序的信息，适用于任意的列表排序。对于两个都是有序的列表，我们可以这样做，
步骤1：记nums1和nums2的索引分别为i和j，初始值为i, j = 0, 0
步骤2：比较nums1[i]和nums2[j]，如果nums1[i]<nums2[j]，则i++；否则，j++
步骤3：循环做步骤2，循环过程中始终取得两个列表中的较小值，将较小值存入新列表nums中
步骤4：循环结束，将两个列表中没有遍历完的列表加入新列表nums中，得到从小到大排列的列表nums

# nums1和nums2的索引
i, j = 0, 0
# 新列表
nums = []
# 遍历，每次将两者中的较小值取出，存入nums
while True:
    # 当nums1和nums2中的一个遍历完，跳出循环
    if (i > m - 1) or (j > n - 1):
        break
    # 将两者中的较小值存入nums，并更新索引
    if nums1[i] < nums2[j]:
        nums.append(nums1[i])
        i += 1
    else:
        nums.append(nums2[j])
        j += 1
# 如果nums1遍历完
if i == m:
    nums.extend(nums2[j:])
# 如果nums2遍历完
else:
    nums.extend(nums1[i:m])

# 将nums值赋给nums1
for k in range(len(nums)):
    nums1[k] = nums[k]

print(nums1)

3. $O(n+m)$+原地排序

参考力扣官方题解，我们发现可以进行原地排序而不用占用额外空间。

思路：建议直接看力扣官方题解，有图更加形象

p1, p = m - 1, len(nums1) - 1
p2 = n - 1

while p1 >= 0 and p2 >= 0:
    if nums1[p1] < nums2[p2]:
        nums1[p] = nums2[p2]
        p2 -= 1
    else:
        temp = nums1[p1]
        nums1[p] = temp
        p1 -= 1
    p -= 1

nums1[:p2+1] = nums2[:p2+1]

print(nums1)