合并k个有序链表

---

题目描述

给定一个包含k个有序链表的数组，将这k个链表合并成一个有序链表，并返回其头结点。要求：时间复杂度O(nlogk)

---

示例

输入：[{1,2},{1,4,5},{6}]
输出：{1,1,2,4,5,6}

---

解题思路

最开始的想法很暴力：按照合并两个链表的方式，进行k-1次的合并，但是由于每次合并l两个列表需要O(n)的时间复杂度，其中n为两个链表的总的结点数，所以计算合并k个链表其时间复杂度会达到O(kN)，空间复杂度为O(1)，不满足题目要求。
后来想到可以将列表中的链表两个两个合并，采用分而治之的思想，这样可以减少结点遍历比较的次数，时间复杂度可满足要求为O(nlogk)，因为排序每个结点大概需*要O(logk)*次合并。

代码示例

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeKLists(self, lists: List[Optional[ListNode]]) -> Optional[ListNode]:
        dummyNode = ListNode(0)
        res = dummyNode
        # 定义间隔，首次间隔为1，接下来每次for循环结束后，间隔变为上一个的2倍
        interval = 1
        # 当合并链表之间的间隔大于k，就结束合并
        amount = len(lists)
        while interval < amount:
            for i in range(0, amount-interval, 2*interval):
            # 进行列表中的两两合并
                lists[i] = self.merge2Lists(lists[i],lists[i+interval])
            interval *= 2
        # 最终返回合并在第一个位置的总链表，同时需要考虑存在k=0的边界情况
        return lists[0] if amount > 0 else None
    # 合并两个链表的函数
    def merge2Lists(self,pHead1,pHead2) -> Optional[ListNode]:
        dummyNode = ListNode(0)
        res = dummyNode
        while pHead1 and pHead2:
            if pHead1.val <= pHead2.val:
                dummyNode.next = pHead1
                pHead1 = pHead1.next
            else:
                dummyNode.next = pHead2
                pHead2 = pHead2.next 
            dummyNode = dummyNode.next
        if pHead1 or pHead2:
            dummyNode.next = pHead1 if pHead1 else pHead2
        return res.next