合并k个有序序列——python

最新推荐文章于 2025-06-24 20:12:32 发布

原创

最新推荐文章于 2025-06-24 20:12:32 发布 · 2.1k 阅读

5 ·

CC 4.0 BY-SA版权

博客内容介绍了如何使用Python的heapq模块简洁地合并多个有序序列。通过讲解heapq.merge函数的工作原理，展示了如何从有序序列中依次取出最小元素，形成新的有序序列。文章以代码示例和步骤解析来简化了课本上复杂的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

课本上给出了合并有序序列的代码，思路简单，直接看课本，上代码：

#coding=utf-8
from collections import namedtuple
import heapq

def mergeKSortedArrays(alist):
    h = list()  # 最小堆
    res= list() # 合并后的输出
    heapContent = namedtuple('contents', ('elem', 'array_idx', 'array_elem_idx'))
    # 每一个序列k的第一个元素按照堆结构组织
    for i, k in enumerate(alist):
        heapq.heappush(h, heapContent(k[0],i,1))
    total_elems = len(alist)* len(alist[0])
    for _ in range(0, total_elems):
        popped = heapq.heappop(h)
        if popped.elem == float("inf"):
            continue
        res.append(popped.elem)   # 将堆中最小元素弹出并加入到res中
        next_array = popped.array_idx
        next_elem_idx = popped.array_elem_idx
        if next_elem_idx < len(alist[next_array]):
            # 将被移除出堆所属的序列的下一个元素插入到当前堆中
            heapq.heappush(h, heapContent(alist[next_array][next_elem_idx], \
                next_array, next_elem_idx+1))
        else:
            # 如果没有元素在当前序列中,则插入一个最大整数
            heapq.heap