Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
题意很明确,将k个有序链表合并成一个有序链表。
首先我们可以考虑简单的情况,即合并两个有序链表,然后使用归并操作即可,
对于合并两个有序链表 Merge 2 Sorted Lists,复杂度是O(n) , n为链表长度
注意,原地归并,不要申请新的链表
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if(l1 == NULL)
return l2 ;
if(l2 == NULL)
return l1 ;
ListNode *t1 = l1 ;
ListNode *t2 = l2 ;
ListNode *th ;
ListNode *tn ;
if(t1->val < t2->val){
th = t1 ;
t1 = t1->next ;
}else{
th = t2 ;
t2 = t2->next ;
}
tn = th ;
while(t1 != NULL && t2 != NULL){
while( t1 != NULL && t1->val <= t2->val){
//cout << t1->val << endl ;
tn->next = t1 ;
tn = t1 ;
t1 = t1->next ;
}
if(t1==NULL)
break ;
while(t2!=NULL && t2->val <= t1->val ){
// cout << t2->val << endl ;
tn->next = t2 ;
tn = t2 ;
t2 = t2->next ;
}
}
if(t1 != NULL)
tn->next = t1 ;
if(t2 != NULL)
tn->next = t2 ;
return th ;
}
随后使用归并,第一轮将k个链表合并为k/2个,第二轮合并为k/4个,直到合并为1个为止。考虑到每次合并后链表长度都变为原来的2倍,尽管每一轮合并次数减半,但整体开销未变,每轮开销为O(kn),一共有logk轮,因此整体复杂度为O(nklogk)。
ListNode* mergeKLists(vector<ListNode*>& lists) {
vector<ListNode*> temp ;
if(lists.size()==0)
return NULL ;
for(int i=0 ;i<lists.size() ;i++)
temp.push_back(lists[i]) ;
while(temp.size()>1){
for(int i=0 ;i<temp.size()-1 ;i=i+2){
ListNode* x = mergeTwoLists(temp[i],temp[i+1]) ;
temp[i/2] = x ;
}
if(temp.size()%2==1)
temp[temp.size()/2] = temp[temp.size()-1] ;
int offset = temp.size()/2 +temp.size()%2;
temp.erase(temp.begin()+offset, temp.end()) ;
}
return temp[0] ;
}