合并K个有序链表

问题描述

给定一个链表数组，每个链表都已经按升序排列在这里插入代码片。
请将所有链表合并到一个升序链表中，返回合并后的链表。
letcode原题

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[
  1->4->5,
  1->3->4,
  2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

解决方案

分解：多个有序链表的排序拆分为两个有序链表的排序。其中，迭代排序方式已经很熟悉，此处展示两个有序链表合并的递归方式。

    def mergeTwoList(self, l1, l2):
        if not l1:
            return l2
        if not l2:
            return l1
        if l1.val < l2.val:
            l1.next = self.mergeTwoList(l1.next, l2)
            return l1
        else:
            l2.next = self.mergeTwoList(l1, l2.next)
            return l2

之后进行多个有序链表的排序。

    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        if not lists:
            return None
        for i in range(1, len(lists)):
            lists[0] = self.mergeTwoList(lists[0], lists[i])
        return lists[0]

但是在性能和时间上不是最优的，多个有序链表的排序可以和归并排序联系在一起，先分后合，分而治之。

    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        if not lists:
            return None
        length = len(lists)
        if length ==1:
            return lists[0]
        left, right = lists[:length //2], lists[length//2:]
        return self.mergeTwoList(self.mergeKLists(left), self.mergeKLists(right))