【算法百题之五十五】合并K个升序链表
大家好,我是Lampard~~
很高兴又能和大家见面了,接下来准备系列更新的是算法题,一日一练,早日升仙!
今天的问题是:合并K个升序链表
示例:
输入: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
输入:lists = [] 输出:[]输入:lists = [[]] 输出:[]
思路:还记得我们合并两个有序链表的题目吗?【算法百题之五十三】合并两个有序链表,如果把这道题拆分成n个合并两个有序链表那么就变得简单起来了。
算法代码:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode();
ListNode* curNode = head;
while (l1 && l2) {
ListNode** tmp = l1->val < l2->val ? &l1 : &l2;
curNode->next = *tmp;
*tmp = (*tmp)->next;
curNode = curNode->next;
}
curNode->next = l1 ? l1 : l2;
return head->next;
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
if (lists.size() == 0) return nullptr;
ListNode* head = lists[0];
for (int i = 1; i < lists.size(); ++i) {
if (lists[i]) head = mergeTwoLists(head, lists[i]);
}
return head;
}
顺便帮大家回顾一下这个ListNode** tmp,指针的指针。我们不能用ListNode* tmp来直接指向l1和l2,因为这样的话会生成一个新的ListNode指针,指向l1或l2的地址。在执行*tmp = (*tmp)->next;的时候,并不能让l1和l2指向next指针。而用ListNode** tmp取l1或l2的引用,那么tmp就不是一个新的地址,而是l1或l2的别名,所以在执行*tmp = (*tmp)->next;的时候,l1或者l2就会指向下一位置。
测试结果: