88. 合并两个有序数组
难度:简单
给你两个有序整数数组 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]
提示:
nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109
解法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.
"""
i = j = 0
nums = []
while i < m or j < n:
if i == m:
nums.append(nums2[j])
j += 1
elif j == n:
nums.append(nums1[i])
i += 1
elif nums1[i] < nums2[j]:
nums.append(nums1[i])
i += 1
else:
nums.append(nums2[j])
j += 1
nums1[:] = nums
解法2:逆向双指针
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
for i in range(len(nums1) - 1, -1, -1):
if m - 1 >= 0 and n - 1 >= 0:
if nums1[m-1] >= nums2[n-1]:
nums1[i] = nums1[m-1]
m -= 1
else:
nums1[i] = nums2[n-1]
n -= 1
elif m - 1 >= 0:
nums1[i] = nums1[m-1]
m -= 1
else:
nums1[i] = nums2[n-1]
n -= 1