LeetCode-88. 合并两个有序数组

给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。

注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。

想法:

  1. 做一个新数组nums3,长度和nums1相同,直接循环判断两个数组,将符合条件的放入到nums3中,nums1和nums2谁还有剩余,谁就全部赋值给nums3,最后需要将nums3中的值赋值给nums1
class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        if m == 0 and n != 0:
            for l in range(len(nums2)):
                nums1[l] = nums2[l]
        elif m != 0 and n!= 0:
            nums3 = [0]*len(nums1)
            i, j, k = 0, 0, 0
            while i <= m-1 and j <= n-1:
                if nums1[i] > nums2[j]:
                    nums3[k] = nums2[j]
                    k += 1
                    j += 1
                else:
                    nums3[k] = nums1[i]
                    k += 1
                    i += 1
            if i == m:
                nums3[i+j:] = nums2[j:] 
            if j == n:
                nums3[i+j:] = nums1[i:m]
            for i in range(len(nums3)):
                nums1[i] = nums3[i]
  1. 上面方法空间耗费大,所以改进一下方法,从nums1的末尾,也就是m+n位依次填入数据,最后剩余的再判断谁有剩余,统一赋值过去
class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        if m == 0 and n != 0:
            for l in range(len(nums2)):
                nums1[l] = nums2[l]
        elif m != 0 and n!= 0:
            sumn = m + n
            i, j = m, n
            while i >= 1 and j >= 1:
                if nums1[i-1] > nums2[j-1]:
                    nums1[sumn-1] = nums1[i-1]
                    i -= 1
                    sumn -= 1
                else:
                    nums1[sumn-1] = nums2[j-1]
                    j -= 1
                    sumn -= 1
            if i == 0:
                nums1[0:sumn] = nums2[0:j]
            if j == 0:
                nums1[0:sumn] = nums1[0:i]
posted @ 2022-03-02 09:02  小Aer  阅读(12)  评论(0编辑  收藏  举报  来源