合併 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];
}
};