（力扣记录）23. 合并K个升序链表

最新推荐文章于 2025-12-03 23:35:37 发布

原创最新推荐文章于 2025-12-03 23:35:37 发布 · 465 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #链表 #算法 #数据结构 #python

力扣算法题目记录专栏收录该内容

14 篇文章

订阅专栏

本文介绍了如何使用Python实现合并K个已排序链表的算法，时间复杂度为O(NKlogK)，空间复杂度为O(K)。主要讲解了Solution类中的mergeKLists方法和merge方法的详细过程。

数据类型：链表

时间复杂度：O(NK logK), N为链表长度

空间复杂度：O(K)

代码实现：

# 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 len(lists) == 0: return None

        while len(lists) > 1:
            merged = []
            for i in range(0, len(lists), 2):
                l1 = lists[i]
                l2 = lists[i + 1] if i + 1 < len(lists) else None
                merged.append(self.merge(l1, l2))
            lists = merged
        return lists[0]

    def merge(self, l1, l2):
        curr = out = ListNode()
        while l1 and l2:
            if l1.val < l2.val:
                curr.next = l1
                l1 = l1.next
            else:
                curr.next = l2
                l2 = l2.next
            curr = curr.next

        if l1 or l2:
            if l1:
                curr.next = l1
            else:
                curr.next = l2

        return out.next