合并两个有序的数组_leetcode 88. 合并两个有序数组

88. 合并两个有序数组 题目

给你两个有序整数数组 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]

代码：

#合并两个数组
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: void Do not return anything, modify nums1 in-place instead.
        """
        nums1[:] = sorted(nums1[:m] + nums2)
        return nums1

if __name__ == '__main__':
    numA = [1, 2, 5, 7, 9]
    numB = [2, 4, 6, 7]
    print(Solution().merge(numA, 5, numB, 4))
    print(Solution().merge([1,2,3,0,0,0], 3, [2,5,6], 3)) #[1,2,2,3,5,6]
    print(Solution().merge([1], 1, [], 0)) #[1]

运行：

方法2：

#88. 合并两个有序数组
#给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。
# 初始化ums1 和 nums2 的元素数量分别为m 和 n 。
# 你可以假设nums1有足够的空间（空间大小大于或等于m + n）来保存 nums2 中的元素。

class Solution:
    def mergeB(self, nums1, m, nums2, n):
        # Make a copy of nums1.
        nums1_copy = nums1[:m]
        nums1[:] = []

        # Two get pointers for nums1_copy and nums2.
        p1 = 0
        p2 = 0

        # Compare elements from nums1_copy and nums2
        # and add the smallest one into nums1.
        while p1 < m and p2 < n:
            if nums1_copy[p1] < nums2[p2]:
                nums1.append(nums1_copy[p1])
                p1 += 1
            else:
                nums1.append(nums2[p2])
                p2 += 1

        # if there are still elements to add
        if p1 < m:
            nums1[p1 + p2:] = nums1_copy[p1:]
        if p2 < n:
            nums1[p1 + p2:] = nums2[p2:]
        return nums1


nums1 = [1, 2, 3, 0, 0, 0]
nums2 = [2, 5, 6]
n1 = Solution()
print(n1.mergeB(nums1, 3, nums2, 3))

运行结果：

写法3和写法4：

#合并两个数组
#思路3. 从大遍历到小，直接操作num1，这个比较喜欢。

#解释一下：两个数组合并后一定使得nums1有效部分变成m+n的长度（这个视角下问题就简单了），那么就比较两个数组的尾巴，
# 谁大谁就被放在n+m-1的位置上，被放的那个数组的当前指针往前挪一个，直到其中一个数组被遍历光，如果nums2还有剩余，就将剩下的全放在nums1的最前面。

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.
        """
        while m>0 and n >0:
            if nums1[m-1] >= nums2[n-1]:
                nums1[m+n-1] = nums1[m-1]
                m = m -1
            else :
                nums1[m+n-1] = nums2[n-1]
                n = n-1
        if n > 0 :
            nums1[:n] = nums2[:n]


#简单粗暴，先合并再排序
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.
        """
        nums1[m:m + n] = nums2[:n]
        nums1.sort()