【Leetcode C++】2_9. [code] 合并K个有序链表

9. [code] 合并K个有序链表

9.1. 题目

Leetcode 23

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

示例:

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

9.2. 思路

方法1.暴力合并：

遍历所有链表，将所有节点的值放到一个数组中。

将这个数组排序，然后遍历所有元素得到正确顺序的值。

用遍历得到的值，创建一个新的有序链表。

方法2.逐一比较：

第1个链表和第2个链表的合并成一个长链表，再将该长链表与第3个链表合并又得到一个长链表，再依此类推。每两个链表的合并参考8 Leetcode 21两个有序链表的合并。

方法3.分治思想：
先将K个链表分两组，再在每组内分两组，不断分治。

9.3. 代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if (lists.size()==0) return nullptr;
        if (lists.size()==1) return lists[0];
        if (lists.size()==2){
            return mergeTwoLists(lists[0],lists[1]);
        }
        int mid = lists.size()/2;

        std::vector<ListNode*> sub1_list;
        std::vector<ListNode*> sub2_list;

        for (int i = 0; i < mid; ++i){
            sub1_list.push_back(lists[i]);
        }

        for (int i = mid; i < lists.size(); ++i){
            sub2_list.push_back(lists[i]);
        }

        ListNode* l1 = mergeKLists(sub1_list);
        ListNode* l2 = mergeKLists(sub2_list);
        return mergeTwoLists(l1,l2);
    }
    
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if (l1 == nullptr){return l2;}
        if (l2 == nullptr){return l1;}
        ListNode dummy(-1);
        ListNode* prehead = &dummy;
        while(l1 != nullptr && l2 != nullptr){
            if(l1->val <= l2->val){
                prehead->next = l1;
                l1 = l1->next;
            }
            else{
                prehead->next = l2;
                l2 = l2->next;
            }
            prehead = prehead->next;
        }
        prehead->next = l1 != nullptr ? l1 : l2;
        return dummy.next;
    }
};