合并两个有序数组三种解法（Python）

最新推荐文章于 2024-07-18 17:42:10 发布

某工程师$

最新推荐文章于 2024-07-18 17:42:10 发布

阅读量3.4k

点赞数

分类专栏：算法 python 文章标签：算法

本文链接：https://blog.csdn.net/weixin_37882382/article/details/124615673

版权

python 同时被 2 个专栏收录

47 篇文章 0 订阅

订阅专栏

算法

40 篇文章 1 订阅

订阅专栏

直接合并然后排序时间复杂度 O((m + n)log(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.
        """
        nums1[m:] = nums2
        nums1.sort()

双指针法

借助额外空间时间复杂度 O(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.
        """
        i, j = 0, 0
        sorted = []
        while i < m or j < n:
            if i == m:
                sorted.append(nums2[j])
                j += 1
            elif j == n:
                sorted.append(nums1[i])
                i += 1
            elif nums1[i] <= nums2[j]:
                sorted.append(nums1[i])
                i += 1
            else:
                sorted.append(nums2[j])
                j += 1
        nums1[:] = sorted

在num1的尾部插入不需要额外空间时间复杂度 O(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.
        """
        i, j = m - 1, n - 1
        tail_index = m + n - 1
        while tail_index >= 0:
            if i == -1:
                nums1[tail_index] = nums2[j]
                j -= 1
            elif j == -1:
                nums1[tail_index] = nums1[i]
                i -= 1
            # 注意下 尾部插入时 大的值先插入
            # 和从头插入新数组时相反
            elif nums1[i] >= nums2[j]:
                nums1[tail_index] = nums1[i]
                i -= 1
            else:
                nums1[tail_index] = nums2[j]
                j -= 1
            tail_index -= 1