Leetcode 88:合并两个有序数组(最详细解决方案!!!)

给定两个有序整数数组 nums1nums2,将 nums2 合并到 nums1使得 num1 成为一个有序数组。

说明:

  • 初始化 nums1nums2 的元素数量分别为 mn
  • 你可以假设 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]

解题思路

这个问题没什么好说的,就是考察归并排序中的merge操作。

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:] = nums2[:n]
        all_list = nums1.copy()
        all_len = m + n

        i = 0
        j = m
        for k in range(all_len):
            if i > m - 1:
                nums1[k] = all_list[j]
                j += 1
            elif j > all_len - 1:
                nums1[k] = all_list[i]
                i += 1
            elif all_list[i] < all_list[j]:
                nums1[k] = all_list[i]
                i += 1
            else:
                nums1[k] = all_list[j]
                j += 1

注意这里有一个陷阱,我们必须使用all_list = nums1.copy(),使用all_list = nums1是不行的。这是因为后者表达的意思是all_list和nums1指向的是同一片内存空间,如果nums1变化的话,all_list也会随之改变。

当然这里有一个更pythonic的做法

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:] = nums2[:n]
        nums1.sort()

该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

猜你喜欢

转载自blog.csdn.net/qq_17550379/article/details/80500807