【Leetcode】合并K个排序链表

题目链接：合并K个排序链表

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

题解：这题的前身是合并两个排序链表。在剑指里有写。可以点击链接查看。。

这个题，最好就是用小顶堆，O(nlog(K))。用c++的优先队列可以解决这个小顶堆。

把每个节点丢进优先队列，然后以出队列的顺序作为新链表顺序

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    //优先队列重载cmp函数
    struct cmp{
        bool operator()(ListNode* a,ListNode* b){
            return a->val > b->val; //小顶堆
        }
    };

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode* ,vector<ListNode*> , cmp> que;
        for(auto head:lists){
            if(head)   que.push(head);
        }

        ListNode* ans = new ListNode(0);
        ListNode* res = ans;

        while(!que.empty()){
            ListNode* p = que.top();
            que.pop();
            if(p->next){
                que.push(p->next);
            }
            res->next = p;
            res = res->next;
        }
        return ans->next;
    }
};