Leetcode 88. 合并两个有序数组 Python实现

最新推荐文章于 2025-05-06 16:47:28 发布

原创最新推荐文章于 2025-05-06 16:47:28 发布 · 243 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#指针 #leetcode #算法 #python #数据结构

知识星球打卡专栏收录该内容

21 篇文章

订阅专栏

本篇博客介绍了如何使用Python解决LeetCode的88题——合并两个有序数组。通过示例详细解释了算法思路，包括初始化、合并过程以及最终的有序数组结果。

给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
合并后排序

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.
        """
        nums1[:]=sorted(nums1[:m]+nums2)

开辟新空间暂存nums1非0部分

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

从后往前

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.
        """
        left=m-1 # nums1左指针，指向最后一个非0元素
        right1=len(nums1)-1 # nums1右指针，指向最后一个元素
        right2=n-1 # nums2左指针，指向最后一个元素
        while left>=0 and right2>=0:
            if nums1[left]>=nums2[right2]:
                nums1[right1]=nums1[left]
                left-=1
                right2-=1
            else:
                nums1[right1]=nums2[right2]
                right2-=1
                right1-=1
       
        nums1[:right2+1]=nums2[:right2+1]