题目描述:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
首先了解一下堆排列的模块函数heapq,heapq函数简介
答题思路:遍历lists,将每一个链表的节点添加到堆中,之后不断从堆中取最小值,添加到新的链表中,最后返回这个列表
具体程序如下:
class Solution(object):
def mergeKLists(self, lists):
import heapq
que = []
for node in (lists):
if node != None :
heapq.heappush(que ,(node.val, node))
dummy_node = ListNode(-1)
cur = dummy_node
while que:
val, node = heapq.heappop(que)
cur.next = node
cur = cur.next
if node.next != None:
heapq.heappush(que, (node.next.val, node.next))
return dummy_node.next