【算法百题之五十五】合并K个升序链表

 【算法百题之五十五】合并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就会指向下一位置。

测试结果：

 

OK，今天的博客就到这里，谢谢大家！！！