23. 合并K个升序链表

23. 合并K个升序链表

原始题目链接：https://leetcode-cn.com/problems/merge-k-sorted-lists/

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

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

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

解题思路：

暴力解法是把N个数放到一个数组里，再一起排序，O(NlogN)。
由于k个链表是有序的，可以使用优先队列来进行优化，优先队列在python中可以使用堆来实现。使用优先队列来存储k个链表的头结点的值，每次比较优先队列中的值的大小，pop出最小值，通过修改指针指向，将上次输出最小值的那个链表的下一个节点进行入队操作，重复循环比较，直到队列为空。

代码实现：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        # 因为k个链表是排好序的，先把所有k个链表的头结点放到优先队列里
        # 优先队列使用堆实现，python的堆是小根堆
        # 堆的元素由元组组成（节点的值， 节点属于第几个链表的索引）
        import heapq
        queue = []
        for index, node in enumerate(lists):
            if node is not None:
                heapq.heappush(queue, (node.val, index))
        
        # 定义一个哨兵节点和当前指针，方便操作，最后返回dumpy的next指针即满足题目要求的头结点
        dumpy = ListNode(-1)
        cur = dumpy

        # 每次比较优先队列中的元素，取出较小的元素作为返回链表的第一个节点
        # 因为是小根堆，堆顶元素就是最小的元素
        # 出队元素后，将出队元素指向所在链表的下一个节点，放入堆中，再进行循环比较，直到堆中元素全部比较完毕
        
        while queue:
            # 获取堆顶元素：最小的节点值和所在链表的索引
            _, index = heapq.heappop(queue)
            # 找到最小元素后，将cur的next指针指向pop出的链表节点
            cur.next = lists[index]
            # 更新cur指针的指向
            cur = cur.next
            # 当前pop出的链表的节点的指针指向该链表的下一个节点，用于入堆操作
            lists[index] = lists[index].next
            if lists[index] is not None:
                heapq.heappush(queue, (lists[index].val, index))
        
        # 最后返回dumpy的next
        return dumpy.next

参考文献：
https://leetcode-cn.com/problems/merge-k-sorted-lists/solution/python-c-you-xian-dui-lie-zui-xiao-dui-onlogk-by-m/
https://blog.csdn.net/weixin_38361153/article/details/93514302