欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

88.合并两个有序数组

程序员文章站 2024-02-24 23:41:46
...

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

说明:

初始化 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]

类似归并排序
三指针

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 = 0
        j = 0
        k = 0
        
        l = list(nums1)
        
        while i < m and j < n:
            
            if l[i] <= nums2[j]:
                nums1[k] = l[i]
                i += 1
            else:
                nums1[k] = nums2[j]
                j += 1
                
            k += 1
            
        if i == m:
            while j < n:
                nums1[k] = nums2[j]
                j += 1
                k += 1
        else:
            while i < m:
                nums1[k] = l[i]
                i += 1
                k += 1