合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思路:是给两个有序链表排序的变形,主要在于为了提高速度,不要一个一个加上去,而是两个变成一个长的,再两个变成一个长的,知道最后成为一个长子串。
C++代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
ListNode* Merge(ListNode* l1,ListNode* l2)
{
ListNode* start = new ListNode(-1);
ListNode* end=start;
while(l1 && l2)
{
if(l1->val < l2->val)
{
end->next=l1;
l1=l1->next;
}
else
{
end->next = l2;
l2=l2->next;
}
end = end->next;
}
if(l1)
end->next = l1;
else if(l2)
end->next = l2;
return start->next;
}
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
int n=lists.size();
if(n == 0) return {};
while(n >1) {
int k=(n+1)/2;
for(int i=0;i<n/2;i++)
{
lists[i]=Merge(lists[i],lists[i+k]);
}
n=k;
}
return lists[0];
}
};