88.合并两个有序数组

2021-08-03  本文已影响0人  年少为云

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

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

示例 1:

输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]

示例 2:

输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]

提示:

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array

Python3

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.
        """
        p = m + n - 1
        m = m - 1
        n = n - 1
        while m >= 0 and n >= 0:
            if nums1[m] > nums2[n]:
                nums1[p] = nums1[m]
                m = m-1
                p = p-1
            else:
                nums1[p] = nums2[n]
                n = n-1
                p = p-1
        if n >= 0:
            nums1[:p+1] = nums2[:n+1]
上一篇 下一篇

猜你喜欢

热点阅读