目录(快速导航)
题目描述
视频讲解 https://www.bilibili.com/video/av66438901/
思路
代码
题目描述:
题目链接:https://leetcode-cn.com/problems/merge-k-sorted-lists/
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
视频讲解
https://www.bilibili.com/video/av66438901/
思路:
归并
将邻近的两个链表进行合并,只需要将两个两个都扫一遍即可。
那么有一个链表数组,我们就两两合并,将合并的链表放入下一次合并的行中,如:list0和list1合并出list0放置到下一行;
如果有剩余的链表,也就是不能组队的,直接放入下一次合并行中。如:list4在第二行,不能组队,就直接放入下一行。l
最后迭代到只有一个链表就结束了。具体看代码。
代码:
class Solution {
public:
// 合并a,b两个链表 这个写法可以记住-》递归
ListNode* mergeTwo(ListNode* a, ListNode* b) {
if(a == NULL) return b;
if(b == NULL) return a;
if(a->val<b->val){
a->next = mergeTwo(a->next,b);
return a;
} else {
b->next = mergeTwo(a,b->next);
return b;
}
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.size() == 0) {
return NULL;
}
if (lists.size() == 1) {
return lists[0];
}
//tmp放置分治法参与下一次合并的链表头
vector<ListNode*> tmp;
for (int i = 0;i < lists.size()/2;i++) {
tmp.push_back(mergeTwo(lists[2 * i], lists[2 * i + 1]));
}
// 查看是否有没有组队的链表
if (lists.size() % 2 == 1) {
tmp.push_back(lists[lists.size() - 1]);
}
// 递归
return mergeKLists(tmp);
}
};