88. 合并两个有序数组
题目
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 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]
这题的思路,我只能想到用两个指针,然后从头开始比对,num2大于num1的就插入过去,但是后来发现根本不行。只能面向答案去反推思路
class Solution:
def merge(self,nums1:List[int],m:int,nums2:List[int],n:int)->None:
s1,s2,end=m-1,n-1,m+n-1
while(s1>=0 and s2>=0):
if nums1[s1]<=nums2[s2]:
nums1[end]=nums2[s2]
s2-=1
else:
nums1[end]=nums1[s1]
s1-=1
end-=1
if s2>=0:
nums1[:s2+1]=nums[:s2+1]
思路是:定义三个尾部指针,s1是num1的尾部指针,s2是num2的尾部指针,end是合并后的尾指针。其中s1是指向nums1存在数字的末尾,就比如示例中【1,2,3,0,0,0】,那么s1就是指向3。然后从后往前遍历,如nums2的元素比nums1的元素大,那么就将nums2放到nums1的末尾,也就是
【1,2,3,0,0,6】,然后大数组的指针往前移动;如果nums1的元素大于nums2的元素,那么就将nums1的元素放到nums1的空位置中,如【1,2,3,0,5,6】的时候,nums2=【2,5,6】,而此时s1还是等于3,s2=2,那么,就变成【1,2,3,3,5,6】,然后再比对,发现s1指向的值和s2指向的值相等,那么就变成【1,2,2,3,5,6】。
测试用例:
def merge(nums1,m,nums2,n):
s1,s2,end=m-1,n-1,m+n-1
while(s1>=0 and s2>=0):
if(nums1[s1]<=nums2[s2]):
nums1[end]=nums2[s2]
s2-=1
else:
nums1[end]=nums1[s1]
s1-=1
end-=1
if (s2>=0):
nums1[:s2+1]=nums[:s2+1]
return nums1
nums=[1,2,3,0,0,0]
result=merge(nums,3,[2,5,6],3)
print(result)