Merge k sorted linked lists

本文介绍了一种使用Python的heapq库来合并K个已排序链表的方法。通过将链表节点值推入最小堆并依次弹出,构建新的排序链表。此算法的时间复杂度为O(nlogn),空间复杂度为O(n)。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Merge k sorted linked lists and return it as one sorted list.

每个列表都已经是有序的 所以本质上是对有k个元素的数组进行排序(且数组内的元素需要变化

heapq 为python的一个实现最小堆的库

# T(nlogn) S(n)
import heapq
class Solution:
    def mergeKLists(self, lists):
        heap = []
        for l in lists:
            while l:
                heapq.heappush(heap, l.val)
                l = l.next
        head = ListNode(val=None)
        cur = head
        while len(heap) != 0:
            cur.next = ListNode(heapq.heappop(heap))
            cur = cur.next
        return head.next
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值