LeetCode 23. 合并 K 个升序链表（困难）

 题目描述  

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

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

示例

示例 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 = [[]]
输出：[]

解法

1.最小堆

解题思路

       因为链表已按升序排列，合并后的第一个节点 first，一定是某个链表的头节点。合并后的第二个节点，可能是某个链表的头节点，也可能是 first 的下一个节点。

        例如有三个链表 1->2->5, 3->4->6, 4->5->6，找到第一个节点 1 之后，第二个节点不是另一个链表的头节点，而是节点 1 的下一个节点 2。

        按照这个过程继续思考，每当我们找到一个节点值最小的节点 x，就把节点 x.next 加入（可能是最小节点）的集合中。

        可以用最小堆实现。初始把所有链表的头节点入堆，然后不断弹出堆中最小节点 x，如果 x.next 不为空就加入堆中。循环直到堆为空。把弹出的节点按顺序拼接起来，就得到了答案。

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        auto cmp = [](const ListNode* a, const ListNode* b) {
            return a->val > b->val;  // 如果返回true，a排在b后面,最小堆
        };

        priority_queue<ListNode*, vector<ListNode*>, decltype(cmp)> pq(cmp);
        
        // 添加非空链表的头节点
        for(auto list : lists) {
            if(list != nullptr) {  
                pq.push(list);
            }
        }

        ListNode* dummy = new ListNode(0); //哨兵
        ListNode* cur = dummy;
        
        while(!pq.empty()) {
            auto node = pq.top();
            pq.pop();
            cur->next = node;
            cur = cur->next;
            
            if(node->next != nullptr) {  // 检查下一个节点是否为空
                pq.push(node->next);
            }
        }
        
        ListNode* ans = dummy->next;
        delete dummy;  
        return ans;
    }
};

• 时间复杂度：O(Llogm)，其中 m 为 lists 的长度，L 为所有链表的长度之和。
• 空间复杂度：O(m)，堆中至多有 m 个元素。