LeetCode_88、合并两个数组（python）

题目：

• 给定两个有序整数数组 nums1 和 nums2，
• 将 nums2 合并到 nums1 中，使得 num1 成为一个有序数组。
• 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
• 假设 nums1 有足够的空间来保存 nums2 中的元素。

例子：

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

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

思路：

用双指针，从前往后遍历：

• 定义指针p1指向数组nums1的开头
• 定义指针p2指向数组nums2的开头
• 不断的比较两个指针对应的元素，将小的元素放入结果数组中

所以最终的时间复杂度是 O(m + n) ,空间复杂度是O(m + n)

代码：

class Solution:
    def merge(self, nums1, m, nums2, n):
        nums1_copy = nums1[:m]            # 将数组nums1复制到数组nums1_copy
        nums1[:] = []                             # 清空数组nums1，用来存放结果
        
        p1 = 0                # 定义数组nums1_copy指针
        p2 = 0               # 定义数组nums2指针

        while p1 < m and p2 < n:             # 任意数组循环遍历完，就跳出
            if nums1_copy[p1] < nums2[p2]:          # 如果nums1_copy的元素小于nums2的元素
                nums1.append(nums1_copy[p1])      # 将小的元素添加到结果数组中
                p1 += 1                    # 指针右移，以便再次比较
            else:
                nums1.append(nums2[p2])                  # 将小的元素添加到结果数组中
                p2 += 1                    # 指针右移，以便再次比较
                
        if p1 < m:            # 如果数组nums1没有遍历完，说明剩下的都是大数，我们直接添加到结果数组中
            nums1[p1+p2:] = nums1_copy[p1:]
        if p2 < n:            # 如果数组nums2没有遍历完，说明剩下的都是大数，我们直接添加到结果数组中
            nums1[p1 + p2:] = nums2[p2:]
        return nums1          # 返回结果

验证：

solution = Solution()
nums1 = [1,2,3,0,0,0]
nums2 = [2,5,6]
m = 3
n = 3
print(solution.merge(nums1,m,nums2,n))

# 结果：
[1, 2, 2, 3, 5, 6]