【LeetCode】每日一题：合并K个升序链表

孑渡

于 2024-06-26 20:11:34 发布

阅读量292

点赞数 1

分类专栏：算法刷题文章标签： leetcode 链表 python 算法职场和发展

本文链接：https://blog.csdn.net/qq_44459787/article/details/139997501

版权

算法刷题专栏收录该内容

144 篇文章 13 订阅

订阅专栏

给你一个链表数组，每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中，返回合并后的链表。

解题思路

分治加两个链表合并，或者用根堆，根堆的初始化方法很值得背诵，还涉及lambda的用法

AC代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
ListNode.__lt__ = lambda a, b: a.val < b.val

class Solution:
    def mergeKLists(self, lists: List[Optional[ListNode]]) -> Optional[ListNode]:
    
        h = [head for head in lists if head]

        dummyhead = ListNode()
        indexhead = dummyhead
        heapify(h)
        while h:
            node = heappop(h)

            if node.next:
                heappush(h, node.next)
            indexhead.next = node
            indexhead = indexhead.next
        return dummyhead.next