Problem :
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
**
Algorithm:
**
通过两个list的归并在推出多个lists的归并,两个list归并是下列代码的mergeTwoLists()函数。做法是:
取任意一个list作为基准,在另外一个list的头指针跟基准list的头指针比较,如果比他小,就插到基准list头指针的前面,否则就更新基准list的头指针。如果基准list的头指针指到了最后的NULL,那么直接把另外一个list接到基准指针后面。
回到本题,知道了两个list如何归并,那么对于多个lists,每次把lists对半分成两个子lists,然后让两个lists归并,再把子lists对半分,最后剩下两个list归并,然后再回溯。
算法时间复杂度:
假设总共有k个list,每个list的最大长度是n,T(k) = 2T(k/2)+O(n*k)。由大师定理,可以算出算法的总复杂度是O(klogk)。
**
Code:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.size()==0)
return NULL;
return find(lists,0,lists.size()-1);
}
ListNode* find(vector<ListNode*>& lists, int l, int r)
{
if(l<r)
{
int m=(l+r)/2;
return mergeTwoLists(find(lists,l,m),find(lists,m+1,r));
}
return lists[l];
}
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* temp=new ListNode(-1);
ListNode* head=new ListNode(-1);
head->next=l1;
ListNode* help=head;
while(l2!=NULL&&l1!=NULL)
{
if(l2->val<l1->val)
{
help->next=l2;
temp=l2->next;
help=l2;
l2->next=l1;
l2=temp;
}
else
{
help=l1;
l1=l1->next;
}
}
if(l1==NULL)
help->next=l2;
return head->next;
}
};