Leetcode刷题笔记27-合并两个有序数组

1. 题目
给定两个有序整数数组 nums1 和 nums2，将 nums2 合并到 nums1 中，使得 num1 成为一个有序数组。

说明:

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

2. 示例

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

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

3. 解答

python3 48ms

内置数据结构，不需要return。

class Solution:
    def merge(self, nums1, m, nums2, n):
        """
        :type nums1: List[int]
        :type m: int
        :type nums2: List[int]
        :type n: int
        :rtype: void Do not return anything, modify nums1 in-place instead.
        """
        p = 0
        for i in range(n):
            while nums1[p] <= nums2[i] and p < m:
                p += 1
            # print(p)
            # print(nums1[p])
            if p >= m:
                nums1[(p+i):] = nums2[i:]
                # nums1.sort()
                break
            else:
                # nums1[n + p - 1] = nums2[i]
                nums1[m+i] = nums2[i]
        nums1.sort()
        # return nums1
solution = Solution()
nums1 = [1,2,3,0,0,0]
nums2 = [2,5,6]
solution.merge(nums1, 3, nums2, 3)
print(nums1)