当前位置:   article > 正文

k个有序列表合并为一个有序列表 Python3版本_k个长度不同的有序表

k个长度不同的有序表
  1. import heapq
  2. def merge(lists):
  3. heap = []
  4. for i,l in enumerate(lists):
  5. heap.append((l.pop(0),i))
  6. heapq.heapify(heap)
  7. result = []
  8. while heap:
  9. val,ids = heapq.heappop(heap)
  10. result.append(val)
  11. if lists[ids]:
  12. heapq.heappush(heap,(lists[ids].pop(0),ids))
  13. return result

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

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/177599
推荐阅读
相关标签
  

闽ICP备14008679号