题目:
合并 k 个排序链表,返回合并后的排序链表
示例
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思路
使用python的优先队列PriorityQueue
实现。python的优先队列是用堆实现的,本质是一个二叉树,进入的元素会自动进行排序。
代码
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
import Queue
class Solution(object):
def mergeKLists(self, lists):
"""
:type lists: List[ListNode]
:rtype: ListNode
"""
q = Queue.PriorityQueue()
head = node = ListNode(0)
if not lists:
return
for l in lists:
if l:
q.put((l.val, l))
# 这个循环很巧妙,保证了找出k个有序链表中最小的数并将该数所在的链表后移。
# 队列里面存的始终是k个有序列表的头。
while not q.empty():
val, temp = q.get()
node.next = ListNode(val)
node = node.next
temp = temp.next
if temp:
q.put((temp.val, temp))
return head.next