LeetCode:23. 合并K个排序链表(python)

LeetCode:23. 合并K个排序链表(python)

合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6

LeetCode 链接

思路

附代码(Python3):
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def mergeKLists(self, lists):
        '''合并K个有序链表(分治法)'''
        if not lists:
            return None
        n = len(lists)
        interval = 1              # 间隔
        while interval < n:     
            for i in range(0, n-interval, interval*2):
                lists[i] = self.merge2Lists(lists[i], lists[i+interval])
            interval *= 2         # 更新间隔
        return lists[0]
    
    def merge2Lists(self, l1, l2):
        '''合并两个有序链表'''
        res = cur = ListNode(None)
        while l1 and l2:
            if l1.val < l2.val:
                cur.next = l1
                l1 = l1.next
            else:
                cur.next = l2
                l2 = l2.next
            cur = cur.next
        if l1:
            cur.next = l1
        else:
            cur.next = l2
        return res.next
def build_link(nums):
    '''创建链表'''
    res = cur = ListNode(None)
    for i in nums:
        cur.next = ListNode(i)
        cur = cur.next
    return res.next

# 创建链表列表
nums_li = [[1, 4, 5], [1, 3, 4], [2, 6]]
lists = []
for i in nums_li:
    lists.append(build_link(i))

# 合并K个排序链表
test = Solution()
res = test.mergeKLists(lists)
while res:
    print(res.val, end=' ')
    res = res.next
print()
1 1 2 3 4 4 5 6 
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值