Leetcode 23. 合并K个升序链表

关注

原创最新推荐文章于 2026-08-06 22:45:46 发布 222 阅读 AI 写同款· GEO 优化 ›

2025年大模型算法工程师春季社招面试经验汇总自我介绍是对背景和工作经历的概述，所以尽可能突出你自己做的比较好且有把握的项目，这样面试官在提问的时候也会因为你自己提到的内容而往这方面走，不过有一些面试官就是会找简历上和自己业务相关或者感兴趣的项目来进行提问，所以你在简历上提到的项目都需要非常熟悉。BertDP （data parallelism）数据并行，数据并行是指将训练数据分成多个小批次，同时在多个计算设备（如 GPU）上进行模型训练，每个设备都拥有完整的模型副本。阅读详情

题目链接：https://leetcode.cn/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 = [[]]
输出：[]

这题显然是数组k路归并的扩展，而数组k路归并存在三种解法（假设k路数组的所有元素数为n）：
（1）k路数组拷贝至同一数组，再统一排序
（2）k路数组进行k次归并
（3）使用最小堆，依次从k个数组中找到最小值，重复n次
而这题把数组换成了链表，那使用第三种方法是比较合适的。

# 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[Optional[ListNode]]) -> Optional[ListNode]:
        if not lists:
            return None
        k = len(lists)
        dummy = ListNode(val=-1)
        tail = dummy
        heap = []
        # put each list head to min heap
        for i, node in enumerate(lists):
            if node is not None:
                heap.append((node.val, i, node))  # 当val相同通过link_index:i比较
        heapq.heapify(heap)
        # find min node n times
        while len(heap) > 0:
            _, idx, cur_node = heapq.heappop(heap)
            tail.next = cur_node
            tail = cur_node
            if cur_node.next is not None:
                heapq.heappush(heap, (cur_node.next.val, idx, cur_node.next))
        return dummy.next