合并K个升序链表

题目描述：

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

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

示例 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
示例 2：

输入：lists = []
输出：[]
示例 3：

输入：lists = [[]]
输出：[]

提示：

k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按 升序 排列
lists[i].length 的总和不超过 10^4

思路:

思路很简单，写一个两个链表合并的函数，每次两个链表合并最后输出

代码:

func mergeKLists(lists []*ListNode) *ListNode {

    if len(lists)==0{
        return nil
    }else if len(lists)<2{ 
        return lists[0]
        
    }
    for i:=1;i<len(lists);i++{
        lists[0] = mergeTwoLists(lists[i],lists[0])
    }
    return lists[0]
}
func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    if l1==nil{
        return l2
    }else if l2==nil{
        return l1
    }
    head,l3 := &ListNode{},&ListNode{}
    if l1.Val>l2.Val{
        head = l2
    }else{
        head = l1
    }
    
    for l1!=nil&&l2!=nil{
        if l1.Val > l2.Val{
            l3.Next=l2
            l3 = l2
            l2 = l2.Next
        }else{
            l3.Next=l1
            l3 = l1
            l1 = l1.Next
        }
    }
    if l1!=nil{
        l3.Next = l1
    }else if l2!=nil{
        l3.Next = l2
    }
    return head
}

代码效率：

执行用时：112 ms, 在所有 Go 提交中击败了32.23%的用户
内存消耗：5.3 MB, 在所有 Go 提交中击败了73.77%的用户