问题描述:
合并两个升序的整数数组 A 和 B 形成一个新的数组,新数组也要有序。
问题实示例:
输入 A = [1],B = [1],输出 [1,1],返回合并后的数组,输入 A = [1,2,3,4],B = [2,4,5,6],
输出[1,2,2,3,4,4,5,6],返回合并所有元素后的数组。
"""
问题描述:
合并两个升序的整数数组 A 和 B 形成一个新的数组,新数组也要有序。
问题实示例:
输入 A = [1],B = [1],输出 [1,1],返回合并后的数组,输入 A = [1,2,3,4],B = [2,4,5,6],
输出[1,2,2,3,4,4,5,6],返回合并所有元素后的数组。
"""
#代码演示:
#方法一:使用 zip()函数
class Zip_way:
def __init__(self):
self.list1 = [1, 2, 3, 4]
self.list2 = [2, 4, 5, 6]
def lists(self):
self.new_list = []
#zip()函数只能将个数相同的列表连接起来,若一个列表比另一个列表个数多,多余的部分将不会出现在连接后的列表里面。
self.zip_list = list(zip(self.list1,self.list2))
for i in self.zip_list:
self.new_list.append(i[0])
self.new_list.append(i[1])
return self.new_list.sort()
zip_way = Zip_way()
zip_way.lists()
print("输入:",zip_way.list1," ",zip_way.list2)
print("合并完成后:",zip_way.new_list)
#方法二:for 循环
class For_way:
def __init__(self,list1,list2):
self.list1=list1
self.list2=list2
def two_for(self):
self.new_list = []
for i in self.list1:
self.new_list.append(i)
for i in self.list2:
self.new_list.append(i)
return self.new_list.sort()
list1 = [1,2,3,4]
list2 = [2,4,5,6,8]
for_way = For_way(list1,list2)
for_way.two_for()
print("输入:",list1," ",list2)
print("合并完成后:",for_way.new_list)
#方法三:
class Solution:
#参数 A:有序整数数组 A
#参数 B:有序整数数组 B
#返回:一个新的有序整数数组
def mergeSortedArray(self,A,B):
i,j = 0,0
C= []
#将两个列表中符合条件的元素按大小添加进C列表中
while i < len(A) and j < len(B):
if A[i] < B[j]:
C.append(A[i])
i += 1
else:
C.append(B[j])
j += 1
#将剩下数字特别大的元素添加到C列表中
while i < len(A):
C.append(A[i])
i += 1
while j < len(B):
C.append(B[j])
j += 1
return C
#主函数
if __name__ == "__main__":
A = [1,4]
B = [1,2,3]
D = [1,2,3,4]
E = [2,4,5,6]
solution = Solution()
print("输入:",A,"",B)
print("输出:",solution.mergeSortedArray(A,B))
print("输入:", D, "", E)
print("输出:", solution.mergeSortedArray(D, E))
结果: