题目描述
这是leetcode上面的一道编程题,题目如下
/*
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
*/
解题思路
利用递归分治的思想将k个有序链表的合并问题,分解成多个合并两个有序链表的问题。比如示例中我们将三个链表分成两份,第一个链表单独为一份,第二个和第三个链表为一份,然后将第二个和第三个按照两个有序链表合并的思路进行合并,之后再将得到的结果和第一个链表进行合并,就可以得到一个有序的链表。
实现代码
ListNode* mergeKLists(vector<ListNode*>& lists, int begin, int end) {
if (end < begin) return NULL;
if (end == begin) return lists[begin];
int mid = (begin + end) / 2;
auto p1 = mergeKLists(lists, begin, mid);
auto p2 = mergeKLists(lists, mid + 1, end);
ListNode head(0);
ListNode* cur = &head;
while (p1 && p2) {
if (p1->val < p2->val) {
cur->next = p1;
p1 = p1->next;
} else {
cur->next = p2;
p2 = p2->next;
}
cur = cur->next;
}
if (p1) cur->next = p1;
else if (p2) cur->next = p2;
return head.next;
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
return mergeKLists(lists, 0, lists.size() - 1);
}