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