天天看点

leetcode Merge k Sorted Lists

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

之前知道这道题目的时间复杂度应该为logk*k*m,k是链表数目,m是每条链表的长度。logk为每次选择最小的元素时间复杂度。看到这个复杂度,首先想到了最小堆,但不想自己实现。就使用STL的map,因为会出现一样的元素所以使用multimap。

/**
 * 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) {
        multimap<int,ListNode *> m;
        for(int i=0;i<lists.size();i++){
            if(lists[i]!=NULL){
                m.insert(make_pair(lists[i]->val,lists[i]));
            }
        }
        ListNode *dummy=new ListNode(INT_MIN);
        ListNode *link=dummy;
        while(!m.empty()){
            map<int,ListNode *>::iterator mi=m.begin();
            link->next=mi->second;
            link=link->next;
            m.erase(mi);
            if(link->next!=NULL){
                m.insert(make_pair(link->next->val,link->next));
            }
        }
        link->next=NULL;
        return dummy->next;
    }
    
};