题目
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input:
[
1->4->5,
1->3->4,
2->6
]
Output: 1->1->2->3->4->4->5->6
分析
这道题的描述十分简单,要求也清晰,就是要我们把k
个有序的链表合并成一个有序的链表,我们熟悉的应该是合并两个链表,但这里链表数目增加了,算法的设计就自然变难了。
合并两个有序链表
我们先回顾经典的合并两个链表的问题,最好的方法如下:
- 从合并两个链表头节点开始,比较两个链表头节点值的大小,若
list
的头节点值更小,则把result_list
的头指针指向这个节点,并把list
的头节点往后移一次 - 循环判断剩余节点:继续比较两个链表头节点大小,若
list
的头节点值更小,则把result_list
尾节点的next
指针指向list
的头节点,并把list
的头节点往后移一次。退出循环的条件是两个链表中有一个头节点指针为空 - 使
result_list
的尾节点的next
指针指向不为空的那个链表的头节点
如果引入一个伪头部,则可以把第1、2步合并,简化代码。
代码如下:
//l1和l2分别指向要合并的两个有序链表的头节点
ListNode* mergeList(ListNode* l1, ListNode* l2) {
ListNode* dummy = new ListNode(-1);
ListNode* node = dummy;
while (l1 && l2) {
if (l1->val < l2->val) {
node->next = l1<