[LeetCode]23. Merge k Sorted Lists

0. 问题地址

https://leetcode.com/problems/merge-k-sorted-lists/

1. 问题简述

给定多(k)个有序的链表，将其合并为1个链表

2. 解题思路

很容易想到优先队列的应用，维护一个大小为k的，以链表节点的值和该节点作为一个二元tuple的优先队列。先将每个链表的头结点放入，每次去出堆顶（最小值）的节点，构造新的链表。然后放入取出的节点的next指向的节点（若已经是尾节点则不放入），重复该步骤直至队列为空，新的链表就已经构造完成。

复杂度应该是O(n*log(k))，n为总节点数，k为初始链表数。

可以使用Queue模块的PriorityQueue，但是效率似乎不如自己手写来得快，这个解法在所有Python解法中谈不上快。

3. 题解代码

#coding=utf-8
# Definition for singly-linked list.
class ListNode(object):
    def __init__(self, x):
        self.val = x
        self.next = None

from Queue import PriorityQueue
class Solution(object):
    def mergeKLists(self, lists):
        q = PriorityQueue()
        head = ListNode(None)
        curNode = head
        for node in lists:
            if node:
                q.put((node.val, node))
        while q.qsize() > 0:
            # 取出堆顶的节点，将新的链表后移
            curNode.next = q.get()[1]
            curNode = curNode.next
            # 如果该节点存在next，则放入堆中
            if curNode.next:
                q.put((curNode.next.val, curNode.next))
        return head.next