完成 merge(L1, L2) 函数:输入参数是两个从小到大排序好的整数列表 L1 和 L2,
高中舍友给我发的题
返回合成后的从小到大排序好的大列表 X
例如 merge([1,4,5], [2, 7, 11]) 会返回 [1,2,4,5,7]; merge([], [2,3,4]) 会返回 [2,3,4]
[1, 5, 7, 9], [2,4,6]
要求:
(1) 程序中比较两个元素大小的次数不能超过 len(L1) + len(L2)
(2) 只能用列表 append() 和 len() 函数
很迷,这道题想了好久,写了好几种解法,然后发布这篇文章之前又改了一次。现在的方案和代码量是我能想到的最优解。
难点在于如何在程序比较两个元素大小的次数不能超过 len(L1) + len(L2) 显然桶排序(o(2*(m+n))、冒泡排序(o(n^2))、快速排序(o(nlogn))统统不能用。好在给定的两个数列都已经排序好了,这里用归并排序。

先来个循环,对两个列表分别计数,每一步都取最小值到新列表,比如取到l1[0] 下一个循环就让 l1[1]跟 l2[0] 比较,以此类推。如果有一个列表全部元素都已经被比较完了并全部加入新列表中了,就直接让另一个列表内的剩下的元素加到新列表后面。
经过这波操作之后,最大比较次数为 len(l1)+len(l2) -1,某些情况下比较次数会降到 1/3[ len(l1)+len(l2) ]
下面是实现的代码
# 时间复杂度 n(n) n < len(1)+len(2) def merge(l1, l2): l1_length = len(l1) l2_length = len(l2) if l1_length is 0: return l2 if l2_length is 0: return l1 x = [] i = 0 j = 0 m = 0 while True: m += 1 if l1[i] < l2[j]: x.append(l1[i]) if i < l1_length - 1: i += 1 else: for s in l2[j:]: x.append(s) break else: x.append(l2[j]) if j < l2_length - 1: j += 1 else: for l in l1[i:]: x.append(l) break print('两个列表的长度和 %s 比较次数 %s' % (l1_length + l2_length, m)) return x
文章最后修订于 2021年6月23日
评论 (0)