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 @   小Aer  阅读(13)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
点击右上角即可分享
微信分享提示