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;
}
};