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]