數組合並演算法(Python)實現
阿新 • • 發佈:2018-12-23
合併兩個有序陣列(列表)
函式功能:傳入兩個有序列表,返回一個合併好的(有序)列表
引數 :兩個有序列表
思路:將長度短的列表的每個元素依次比較(先比較兩個list的長度,參len(list1) >= len(list2),如果 list2 的某個值大於等於 list1 中的某個值並且小於 list1 中的後一個值,就插入並退出進行下一次比較
from copy import deepcopy
def main(list1, list2, recursion = False):
''' 合併兩個列表 返回一個新列表 '''
if recursion:
new_list = [] #新建一個列表,用於返回
i = 0
for iters in list2: # 遍歷 list2
while i < len(list1): #遍歷 list1
if iters >= list1[i] and iters <= list1[i + 1]: # 如果 list2 的某個值大於等於 list1 中的某個值並且小於 list1 中的後一個值
list1.insert(i + 1, iters) # 插入
break
elif iters <= list1[i]: # 如果 list2 中的某個值小於等於 list1 中的某個值
list1.insert(i, iters) # 插入
break
i += 1
new_list = deepcopy(list1)
return new_list
else:
if len(list1) >= len(list2):
return main(list1, list2, recursion = True)
if len(list1) <= len(list2):
list1, list2 = list2, list1
return main(list1, list2, recursion = True )
if __name__ == '__main__':
arr1 = [1,5,9,77]
arr2 = [5,6]
arr3 = [4,9,55]
arr4 = [5,8,56]
arr5 = [1,2,9]
arr6 = [5,6,7,11]
print(main(arr1, arr2))
print(main(arr4, arr3))
print(main(arr5, arr6))