[分治] leetcode 23 合并K个升序链表
1.题目
题目链接
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例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
2.分析
2.1.问题分解
在思考如何解决合并K个升序链表之前,考虑如何合并两个升序链表。
合并两个链表的过程有点类似于归并排序中并的过程,利用两个指针分别遍历两个链表,比较当前两个节点中数的大小,将较小的数加入新链表即可。
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode(-1);
ListNode* cur = head;
while(l1 != nullptr && l2 != nullptr) {
if(l1->val < l2->val) {
cur->next = l1;
l1 = l1->next;
}else{
cur->next = l2;
l2 = l2->next;
}
cur = cur->next;
}
cur->next = l1 == nullptr ? l2 : l1;
return head->next;
}
2.2.简单改进
既然我们现在已经知道了如何合并两个链表,那么我们可以令结果链表为a,每次从链表数组中获取一个链表b,不断将a、b合并即可:
ListNode* mergeKLists(vector<ListNode*>& lists) {
ListNode *ans = nullptr;
for (int i = 0; i < lists.size(); i++) {
ans = mergeTwoLists(ans, lists[i]);
}
return ans;
}
2.3.分治法
用分治法优化后的代码:
ListNode* merge(vector <ListNode*> &lists, int l, int r) {
if (l == r) {
return lists[l];
}
if (l > r) {
return nullptr;
}
int mid = (l + r) / 2;
return mergeTwoLists(merge(lists, l, mid), merge(lists, mid + 1, r));
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
return merge(lists, 0, lists.size() - 1);
}
3.代码
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode(-1);
ListNode* cur = head;
while(l1 != nullptr && l2 != nullptr) {
if(l1->val < l2->val) {
cur->next = l1;
l1 = l1->next;
}else{
cur->next = l2;
l2 = l2->next;
}
cur = cur->next;
}
cur->next = l1 == nullptr ? l2 : l1;
return head->next;
}
ListNode* merge(vector <ListNode*> &lists, int l, int r) {
if (l == r) {
return lists[l];
}
if (l > r) {
return nullptr;
}
int mid = (l + r) / 2;
return mergeTwoLists(merge(lists, l, mid), merge(lists, mid + 1, r));
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
return merge(lists, 0, lists.size() - 1);
}
};