leetcode 23. 合并K个排序链表

题目描述：

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

 

思路分析：

利用辅助空间求解，用最大堆来存链表中的每个节点，再一次一次从堆中取出，建表。

需要注意输入的链表可能全为空，所以建完堆后，需要先判断一次堆是否为空，若为空，则直接输出空。

 

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)
            return nullptr;
        priority_queue<int> q;
        int k=lists.size();
        for(int i=0; i<k; i++)
        {
            ListNode* cur = lists[i];
            while(cur)
            {
                q.push(cur->val);
                cur = cur->next;
            }
        }
        if(q.empty())
            return nullptr;
        // ListNode* res = new ListNode(0);
        ListNode* pre = new ListNode(q.top());
        q.pop();
        while(!q.empty())
        {
            ListNode* tmp = new ListNode(q.top());
            q.pop();
            tmp->next = pre;
            pre = tmp;
        }
        return pre;
    }
};