【leetcode】23 合并K个排序链表（链表，分治）

题目链接：https://leetcode-cn.com/problems/merge-k-sorted-lists/

题目描述

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

示例:

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

思路

1 逐一两两合并有序链表

将合并 k 个链表的问题转化成合并 2 个链表 k-1 次。这里是 21 合并两个有序链表 的题目。
复杂度分析

时间复杂度： $O(kN)$，其中$K$ 是链表的数目。

我们可以在$O(n)$ 的时间内合并两个有序链表，其中 n 是两个链表的总长度。

复杂度分析

• 时间复杂度
  把所有合并过程所需的时间加起来，我们可以得到：
  $O({\sum }_{i=1}^{k-1}(i\ast (\frac{N}{k})+\frac{N}{k}))=O(kN)$

• 空间复杂度：O(1)

我们可以在 O(1)O(1) 空间内合并两个有序链表。

作者：LeetCode
链接：https://leetcode-cn.com/problems/two-sum/solution/he-bing-kge-pai-xu-lian-biao-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *head = nullptr;
        if(lists.empty()) return head;
        for (auto list:lists) {
            head = mergeTwoLists(head, list);
        }
        return head;
    }

    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1 == nullptr) return l2;
        if(l2 == nullptr) return l1;
        if(l1->val < l2->val){
            l1->next = mergeTwoLists(l1->next, l2);
            return l1;
        }else{
            l2->next = mergeTwoLists(l1, l2->next);
            return l2;
        }
    }
};

2 分治

这个方法沿用了上面的解法，但是进行了较大的优化。我们不需要对大部分节点重复遍历多次。

• 将k个链表配对并将同一对中的链表合并
• 第一轮合并后，k个链表合并成$\frac{k}{2}$个链表，平均长度为 $\frac{2N}{k}$ ，然后是$\frac{k}{4}$个链表，$\frac{k}{8}$个链表等等。

因此，我们在每一次配对合并的过程中都会遍历几乎全部 N个节点，并重复这一过程 ${\mathrm{l}\mathrm{o}\mathrm{g}}_{2}K$次。

/*
 * 分治；k个链表配对并将同归对中的链表合并-两两合并，不需要对大部分节点重复遍历多次。
 * 时间复杂度O(nlogk) 空间复杂度O(1)
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *head = nullptr;
        if(lists.empty()) return head;
        int len = lists.size();
        int interval = 1;   // 当前合并后的原链表数
        while (interval <= len){
            for (int i = 0; i < len - interval; i+=interval*2)
                lists[i] = mergeTwoLists(lists[i], lists[i+interval]);
            interval *= 2;
        }
        return head;
    }

    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1 == nullptr) return l2;
        if(l2 == nullptr) return l1;
        if(l1->val < l2->val){
            l1->next = mergeTwoLists(l1->next, l2);
            return l1;
        }else{
            l2->next = mergeTwoLists(l1, l2->next);
            return l2;
        }
    }
};