88. 合并两个有序数组
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
这道题也是双指针的题,和之前多的这道题一模一样,所以代码也一样
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.
"""
tem_list = nums1[:m] + nums2
nums1[:] = [] #需要注意是这里,看清题目,返回的是nums1,所以这里赋值一下
start_1, start_2 = 0,m
while start_1 < m or start_2 < m+n:
if start_1 == m:
nums1.append(tem_list[start_2])
start_2+=1
elif start_2 == m+n:
nums1.append(tem_list[start_1])
start_1+=1
elif tem_list[start_1] <= tem_list[start_2]:
nums1.append(tem_list[start_1])
start_1+=1
else:
nums1.append(tem_list[start_2])
start_2+=1
(写完之后感觉给搞复杂了),可以直接不用把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.
"""
tem = nums1[:m]
nums1[:] = []
i,j = 0, 0
while i < m or j < n:
if i == m:
nums1.append(nums2[j])
j+=1
elif j == n:
nums1.append(tem[i])
i+=1
elif tem[i] <= nums2[j]:
nums1.append(tem[i])
i+=1
else:
nums1.append(nums2[j])
j+=1