假设我们有两个排序数组A和B。我们必须将它们合并并仅形成一个排序数组C。列表的大小可能不同。
例如,假设A = [1,2,4,7]和B = [1,3,4,5,6,8],则合并列表C将为[1,1,2,3,4, 4,5,6,7,8]
为了解决这个问题,请遵循以下步骤-
定义i:= 0,j:= 0,结束:= A – 1的长度
而end> = 0而不是A [end],
结束:=结束– 1
而j <B的长度
如果i>结束而不是A [i],则A [i]:= B [j],并将j增加1
否则,如果A [i]> B [j],则执行shift(A,i),A [i]:= B [j],将end和j加1
使我增加1
shift方法将如下所示工作-
接受输入num_arr,然后我
j:= num_arr的长度– 1
而不是num_arr [j]做j:= j – 1
当j> = i时,做num_arr [j + 1] = num_arr [j],而j:= j – 1
让我们看一下实现以获得更好的理解
class Solution(object): def merge(self, nums1, m, nums2, n): i = 0 j = 0 end = len(nums1)-1 while end>=0 and not nums1[end]: end-=1 while j<len(nums2) : if i>end and not nums1[i]: nums1[i] = nums2[j] j+=1 elif nums1[i]>nums2[j]: self.shift(nums1,i) nums1[i] = nums2[j] end+=1 j+=1 i+=1 return nums1 def shift(self,num,i): j = len(num)-1 while not num[j]: j-=1 while j>=i: num[j+1] = num[j] j-=1 ob = Solution()print(ob.merge([1,2,3,0,0,0],3,[2,5,6],3))
[1,2,3,0,0,0] [2,5,6]
输出结果
[1, 2, 2, 3, 5, 6]