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]

思路

由于最后的结果保存在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.

        """

        p1 = m - 1

        p2 = n - 1

        p = m + n - 1

        while p1 >= 0 and p2 >= 0:

            if nums1[p1] > nums2[p2]:

                nums1[p] = nums1[p1]

                p1 -= 1

            else:

                nums1[p] = nums2[p2]

                p2 -= 1

            p -= 1

        if p2 >=0:

            nums1[:p2 + 1] = nums2[:p2 + 1]

巴特西

leetcode刷题-88.合并两个有序数组

题目

思路

实现

最新文章

热门文章