Leetcode 23. 合并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

思路：依次合并两个链表

class Solution {
public:
    ListNode* mergeTwoList(ListNode* l1, ListNode* l2){
        ListNode* dumyhead=new ListNode(0);
        ListNode* tail=dumyhead;
        while (l1!=nullptr && l2!=nullptr){
            if (l1->val<=l2->val){
                tail->next=l1;
                l1=l1->next;
            }
            else{
                tail->next=l2;
                l2=l2->next;
            }
            tail=tail->next;
        }
        if (l1){
            tail->next=l1;
        }
        else{
            tail->next=l2;
        }
        return dumyhead->next;
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode* ans=nullptr;
        for (auto list:lists){
            ans=mergeTwoList(ans,list);
        }
        return ans;
    }
};