题目:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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 = [[]] 输出:[]
题解:
可以使用分治的方法来解决。
思路是比较两个链表的头节点的值,将较小的头节点作为合并后链表的头节点,然后递归地合并剩下的部分。将所有链表两两合并,得到新的链表数组。继续重复这个过程,直到链表数组中只剩下一个链表。最后剩下的一个链表即为合并后的链表。返回该链表即可。
时间复杂度为 O(Nlogk),其中 N 为所有链表的节点总数,k 为链表的数目。在每一轮合并过程中,需要遍历所有的节点。
代码如下:
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1,ListNode* l2) {
if(!l1) {
return l2;
}
else if(!l2) {
return l1;
}
else if(l1->val < l2->val) {
l1->next = mergeTwoLists(l1->next,l2);
return l1;
}
else {
l2->next = mergeTwoLists(l2->next,l1);
return l2;
}
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.size() == 0) {
return nullptr;
}
else if(lists.size() == 1) {
return lists[0];
}
while(lists.size()>1) {
lists.push_back(mergeTwoLists(lists[0],lists[1]));
lists.erase(lists.begin());
lists.erase(lists.begin());
}
return lists[0];
}
};