合并K个排序链表—leetcode23

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

示例:

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

 

思路： 采用最小堆的思想，将K个链表的头节点放入优先队列中，遍历时返回队首元素，并把该队首元素的下一节点放入优先队列中，直至优先队列为空。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    struct compare{
        bool operator()(const ListNode* l1,const ListNode* l2){
            return l1->val > l2->val;
        }
    };
    
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>,compare> queue;
        for(int i=0;i<lists.size();++i){
            if(lists[i]!=nullptr){
                queue.push(lists[i]);
            }
        }
        ListNode* head = new ListNode(0);
        ListNode* temp = head;
        while(!queue.empty()){
            ListNode* top = queue.top();
            queue.pop();
            temp->next = top;
            temp = temp->next;
            if(top->next!=nullptr){
                queue.push(top->next);
            }
        }
        return head->next;
    }
};