1. 程式人生 > >數組合並演算法(Python)實現

數組合並演算法(Python)實現

合併兩個有序陣列(列表)

函式功能:傳入兩個有序列表,返回一個合併好的(有序)列表
引數 :兩個有序列表
思路:將長度短的列表的每個元素依次比較(先比較兩個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))