【LeetCode】23. 合并K个升序链表(归并、链表、堆)

该题是经典的二路归并的扩展，做该题之前必须要完成 21. 合并两个有序链表 。如果合并两个有序链表都完不成，就不要继续往下看了。

题目

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1：

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

示例 2：

输入：lists = []
输出：[]

示例 3：

输入：lists = [[]]
输出：[]

提示：

堆、k个升序链表合并

朴素的做法是k个指针指向k个节点的头节点，每次取k个指针的最小的val合并。什么？每次都是取最小？耳熟？利用数据结构堆。
我们用一个堆，每次将k个升序链表的头节点都插入到堆中，按照结点的数据域val来排序，每次取最小val的结点进行归并，再将该结点后移，插入堆中。反复执行~
时间复杂度O(KN*logk) ,空间复杂度O(K)，k个链表，每个链表n个结点，插入操作是logk，堆中至多有k个结点。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    struct CMP{
        bool operator() (ListNode* a ,ListNode* b){
            return a->val > b->val;
        }
    };
    
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        auto dummy = new ListNode(-1);
        auto tail = dummy;
        priority_queue<ListNode*,vector<ListNode*>,CMP> heap;

        //将k个链表头插入堆
        for (auto p : lists) if (p) heap.push(p);

        while (heap.size()){
            //每次取出最小的节点归并
            auto p = heap.top();
            heap.pop();
            tail = tail->next = p;
            //跳到下一个节点插入堆
            if (p->next) heap.push(p->next);
        }

        return dummy->next;
    }
};