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

2021-03-12 12:27

阅读:808

标签:如何   EAP   return   image   技术   对比   代码   mic   处理   

一日一技:在 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] 

运行效果如下图所示:
技术图片

不过,你并不需要在工作中写出这样的代码,因为 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)

运行效果如下图所示:

技术图片

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

标签:如何   EAP   return   image   技术   对比   代码   mic   处理   

原文地址:https://blog.51cto.com/15023263/2559059


评论


亲,登录后才可以留言!