一日一技：在 Python 里面如何合并多个有序列表并使得结果依然有序？
摄影：产品经理
厨师：kingname
已知两个列表：[1,3,6,7,9]和 [2,4,5,8,10]如何合并两个列表，并得到最终结果 [1,2,3,4,5,6,7,8,9,10]？
最常想到的办法是先把两个列表加到一起，再排序：
a = [1, 3, 6, 7, 9]
b = [2, 4, 5, 8, 10]
c = a + b
c.sort()
print(c)运行效果如下图所示：
但这样一来，你就浪费了这两个列表原本有序这个前置条件。
由于这两个列表有序，所以正确的处理算法应该是这样的：
- 首先对比
a[0]和b[0]，由于a[0]更小，输出a[0]。 - 再对比
a[1]和b[0]，发现b[0]更小，输出b[0]。 - 再对比
a[1]和b[1]，发现a[1]更小，所以输出a[1]。 - ……
整个过程用 Python 来描述，代码如下：
def merge(a, b):
if not a or not b:
yield from (a + b)
return
if a[0] <= b[0]:
yield a[0]
yield from merge(a[1:], b)
else:
yield b[0]
yield from merge(a, b[1:])
list_a = [1, 3, 6, 7, 9]
list_b = [2, 4, 5, 8, 10]
result = list(merge(list_a, list_b))
print(result)运行效果如下图所示：
不过，你并不需要在工作中写出这样的代码，因为 Python 已经为你提供了现成的模块：heapq.merge。使用方法如下：
import heapq
list_a = [1, 3, 6, 7, 9]
list_b = [2, 4, 5, 8, 10]
result = list(heapq.merge(list_a, list_b))
print(result)运行效果如下图所示：
未闻Code
PYTHON干货日更
本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2019-09-19，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
