leetcode 每日一题 88. 合并两个有序数组

双指针

思路:

方法①:从前向后,将nums1中前m个元素取出放到新数组中,对新数组和nums2分别从头开始遍历,在nums1中插入两者中较小元素,较小元素的数组后移一位,较大元素的数组不变,继续判断,直到其中一个数组元素全部插入到num1中,将还有的剩余的数组的剩下所有元素插入到num1中即可。

方法②:从后向前,由于方法①需要用到m的额外空间,可以直接从num1的尾部开始插入,从nums1的第m个元素和nums2的第n个元素开始判断,较大元素插入到nums1的尾部,较大元素的数组前移一位,较小元素的数组不变。

代码1:

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_copy = nums1[:m] 
        nums1[:] = []
        p1 = 0 
        p2 = 0
        while p1 < m and p2 < n: 
            if nums1_copy[p1] < nums2[p2]: 
                nums1.append(nums1_copy[p1])
                p1 += 1
            else:
                nums1.append(nums2[p2])
                p2 += 1
        if p1 < m: 
            nums1[p1 + p2:] = nums1_copy[p1:]
        if p2 < n:
            nums1[p1 + p2:] = nums2[p2:]

代码2:

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] = nums2[p2]
                p2 -= 1
            else:
                nums1[p] =  nums1[p1]
                p1 -= 1
            p -= 1
        nums1[:p2 + 1] = nums2[:p2 + 1]

 

posted @ 2020-06-17 08:47  nil_f  阅读(134)  评论(0编辑  收藏  举报