数组合并算法(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))

猜你喜欢

转载自blog.csdn.net/qq_42265536/article/details/85108191