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