K个升序链表合并 （每日一题）

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

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

 

示例 ：

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

解决思路：首先想到的是在两个有序链表合并的基础上进行改进，于是乎，第一版代码出炉：

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func mergeKLists(lists []*ListNode) *ListNode {
    if len(lists)==0{
        return nil
    }
    if len(lists) == 1{
        return lists[0]
    }
    if len(lists) == 2{
        return mergeTwoLists(lists[0],lists[1])
    }
    mid := len(lists)/2
    lists1 := lists[:mid]
    lists2 := lists[mid:]
    return mergeTwoLists(mergeKLists(lists1),mergeKLists(lists2))
}
func mergeTwoLists(l1,l2 *ListNode) *ListNode{
    if l1==nil{return l2}
    if l2==nil{return l1}
    pre := &ListNode{Val:0,Next:nil}
    cur := pre
        if l1.Val>l2.Val{
            cur.Next = l2
            cur.Next.Next = mergeTwoLists(l2.Next,l1)
        }else{
            cur.Next = l1
            cur.Next.Next = mergeTwoLists(l1.Next,l2)
        }
    return pre.Next
}

　想一下，这个算法的时间复杂度是多少呢？O(NlogN)，

改进一波：

使用堆来进行优化