合并Python中的排序数组
假设我们有两个排序数组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]