天天看點

數組合并算法(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))