LeetCode20—— 合并两个有序数组

LeetCode20—— 合并两个有序数组

前言：

---

题目内容：

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

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n，这样它就有足够的空间保存来自 nums2 的元素。
示例 1：
输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出：[1,2,2,3,5,6]

示例 2：
输入：nums1 = [1], m = 1, nums2 = [], n = 0
输出：[1]

提示：

nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-sorted-array

---

python实现：

方法一：

class Solution(object):
    def merge(self, nums1, m, nums2, n):
        """
        :type nums1: List[int]
        :type m: int
        :type nums2: List[int]
        :type n: int
        :rtype: None Do not return anything, modify nums1 in-place instead.
        """
        i = m - 1  # 指针i用于存放nums1从右到左的指针
        j = n - 1  # 指针j用于存放nums2从右到左的指针
        pos = m + n - 1  # 指针pos用于存储比较i和j指针指向树枝的较大值

        while i >= 0 and j >= 0:
            # 如果nums1的最右边参数小于等于nums2最右边参数，将nums2最右边的值存储到pos所指向的位置，j指针和pos指针都向左移一位
            if nums1[i] <= nums2[j]:
                nums1[pos] = nums2[j]
                pos -= 1
                j -= 1
            # 如果nums1的最右边参数大于nums2最右边参数，将nums1最右边的值存储到pos所指向的位置，i指针和pos指针都向左移一位
            else:
                nums1[pos] = nums1[i]
                pos -= 1
                i -= 1
        # 如果最后nums1的i指针没有遍历到0，因为nums1是有序的。并且此时nums2的参数都已挪到了nums1，所以不用处理
        # 如果最后nums2的j指针没有遍历到0，则需要将nums2剩余的参数按顺序填充到nums1开头
        if j >= 0:
            for z in range(0, len(nums2[:j + 1])):
                nums1[z] = nums2[z]

        # 返回nums1则为排序后的结果
        return nums1


if __name__ == '__main__':
    nums1 = [1,2,3,0,0,0]
    m = 3
    nums2 = [2,5,6]
    n = 3
    aa = Solution()
    print(aa.merge(nums1, m, nums2, n))

方法二：

class Solution(object):
    def merge(self, nums1, m, nums2, n):
        """
        :type nums1: List[int]
        :type m: int
        :type nums2: List[int]
        :type n: int
        :rtype: None Do not return anything, modify nums1 in-place instead.
        """  
        nums1[m:] = nums2
        nums1.sort()
        return nums1

if __name__ == '__main__':
    nums1 = [1,2,3,0,0,0]
    m = 3
    nums2 = [2,5,6]
    n = 3
    aa = Solution()
    print(aa.merge(nums1, m, nums2, n))
    

---