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

最新推荐文章于 2022-04-28 21:09:39 发布

章小幽

最新推荐文章于 2022-04-28 21:09:39 发布

阅读量386

点赞数 1

分类专栏：其他

本文链接：https://blog.csdn.net/qq_36309480/article/details/99697974

版权

其他专栏收录该内容

12 篇文章 0 订阅

订阅专栏

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为三个列表的总长度。