#0.题目描述
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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
#1.解法1
我们前面做过合并2个有序链表,这次也可以借鉴。
注意不用非得用new的方式在堆上建立哑结点,直接用栈分配也是可以的。空指针赋值:nullptr。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
int listLen = lists.size();
ListNode *last = nullptr;
for (int i=0; i<listLen; i++)
last = mergeTwoLists(last, lists[i]);
return last;
}
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode tail;;
ListNode *head = &tail;
while (l1 && l2) {
if (l1->val < l2->val) {
head->next = l1;
l1 = l1->next;
}
else {
head->next = l2;
l2 = l2->next;
}
head = head->next;
}
head->next = (!l1) ? l2 : l1;
return tail.next;
}
};
#2.解法2
上面的方法,我们对每一组数据都要比较一次,我们能否两两配对,先比较,再两两配对,这样比较的次数就是n/2+n/4+……
就ln(n)了,但是怎么存储生成的新节点呢,用数组?
不,受某题的启发,我们用双端队列,前面出,后面进。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
ListNode *fir = nullptr;
ListNode *sec = nullptr;
deque<ListNode *> deList(lists.begin(), lists.end());
if (!deList.size()) return fir;
while (deList.size() != 1) {
fir = deList.front(); deList.pop_front();
sec = deList.front(); deList.pop_front();
deList.push_back(mergeTwoLists(fir, sec));
}
return deList[0];
}
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode tail;;
ListNode *head = &tail;
while (l1 && l2) {
if (l1->val < l2->val) {
head->next = l1;
l1 = l1->next;
}
else {
head->next = l2;
l2 = l2->next;
}
head = head->next;
}
head->next = (!l1) ? l2 : l1;
return tail.next;
}
};