前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >一日一技:在 Python 里面如何合并多个有序列表并使得结果依然有序?

一日一技:在 Python 里面如何合并多个有序列表并使得结果依然有序?

作者头像
青南
发布2019-09-23 17:41:34
1.3K0
发布2019-09-23 17:41:34
举报
文章被收录于专栏:未闻Code

摄影:产品经理

厨师:kingname

已知两个列表:[1,3,6,7,9][2,4,5,8,10]如何合并两个列表,并得到最终结果 [1,2,3,4,5,6,7,8,9,10]

最常想到的办法是先把两个列表加到一起,再排序:

代码语言:javascript
复制
a = [1, 3, 6, 7, 9]
b = [2, 4, 5, 8, 10]
c = a + b
c.sort()
print(c)

运行效果如下图所示:

但这样一来,你就浪费了这两个列表原本有序这个前置条件。

由于这两个列表有序,所以正确的处理算法应该是这样的:

  1. 首先对比 a[0]b[0],由于 a[0]更小,输出 a[0]
  2. 再对比 a[1]b[0],发现 b[0]更小,输出 b[0]
  3. 再对比 a[1]b[1],发现 a[1]更小,所以输出 a[1]
  4. ……

整个过程用 Python 来描述,代码如下:

代码语言:javascript
复制
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。使用方法如下:

代码语言:javascript
复制
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 删除

本文分享自 未闻Code 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档