k个有序列表合并为一个有序列表 Python3版本

多列表合并算法

最新推荐文章于 2024-07-30 23:25:53 发布

原创最新推荐文章于 2024-07-30 23:25:53 发布 · 446 阅读

0 ·

CC 4.0 BY-SA版权

其他专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用堆数据结构高效合并多个已排序列表的算法，时间复杂度为nlogk，其中n为列表元素总数，k为列表数量。通过Python实现，展示了如何利用heapq模块进行操作。

import heapq
def merge(lists):
    heap = []
    for i,l in enumerate(lists):
        heap.append((l.pop(0),i))
    heapq.heapify(heap)
    result = []
    while heap:
        val,ids = heapq.heappop(heap)
        result.append(val)
        if lists[ids]:
            heapq.heappush(heap,(lists[ids].pop(0),ids))
    return result

时间复杂度为nlogk，n为三个列表的总长度。