# LeetCode Merge k Sorted Lists

来源:互联网 发布:数据解放生产力 编辑:程序博客网 时间:2024/06/16 17:59

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] ;    }
0 0
原创粉丝点击