23. 合并K个升序链表（hard）

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

 

示例 1：

输入：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

 

方法一：用容量为K的最小堆优先队列，把链表的头结点都放进去，然后出队当前优先队列中最小的，挂上链表，然后让出队的那个节点的下一个入队，再出队当前优先队列中最小的，直到优先队列为空。时间复杂度O(kn x logk)，空间O(k)

 方法二：用一个数组存每个节点的值，然后排序，建一个新链表

 方法三：分治法，用二分法合并两个有序链表，时间复杂度O(kn x logk)，空间O(log k)

class Solution {
private:
    //合并两个有序链表
    ListNode* mergeTwoLists(ListNode* head1, ListNode* head2)
    {
        if(!head1 || !head2) return head1 == NULL ? head2 : head1;
        ListNode* res = head1;
        if(head1->val < head2->val)
            head1 = head1->next;
        else
        {
            res = head2;
            head2 = head2->next;
        }
        ListNode* cur = res;
        while(head1 && head2)
        {
            if(head1->val < head2->val)
            {
                cur->next = head1;
                head1 = head1->next;
            }
            else
            {
                cur->next = head2;
                head2 = head2->next;
            }
            cur = cur->next;
        }
        cur->next = (head1 == NULL ? head2: head1);
        return res;
    }

    // 二分链表数组
    ListNode* merge(vector<ListNode*>& lists, int left, int right)
    {
        if(left == right) return lists[left];
        if(left > right) return NULL;
        int mid = left + (right - left) / 2;
        return mergeTwoLists(merge(lists, left, mid), merge(lists, mid + 1, right));
    }
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size() == 0) return NULL;
        return merge(lists, 0, lists.size()-1);     
    }
};