Created
February 14, 2013 20:56
-
-
Save zhoutuo/4956339 to your computer and use it in GitHub Desktop.
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/** | |
* Definition for singly-linked list. | |
* struct ListNode { | |
* int val; | |
* ListNode *next; | |
* ListNode(int x) : val(x), next(NULL) {} | |
* }; | |
*/ | |
struct node { | |
int index; | |
ListNode* n; | |
bool operator < (const node& other) const { | |
return n->val >= other.n->val; | |
} | |
}; | |
class Solution { | |
public: | |
ListNode *mergeKLists(vector<ListNode *> &lists) { | |
// Start typing your C/C++ solution below | |
// DO NOT write int main() function | |
if(lists.size() == 0) { | |
return NULL; | |
} | |
vector<node> heap; | |
ListNode result(-1); | |
ListNode* nextNode = &result; | |
for(int i = 0; i < lists.size(); ++i) { | |
ListNode*& cur = lists[i]; | |
if(cur != NULL) { | |
node tmp; | |
tmp.n = cur; | |
tmp.index = i; | |
heap.push_back(tmp); | |
cur = cur -> next; | |
} | |
} | |
make_heap(heap.begin(), heap.end()); | |
while(!heap.empty()) { | |
pop_heap(heap.begin(),heap.end()); | |
node next = heap.back(); | |
heap.pop_back(); | |
next.n -> next = NULL; | |
nextNode->next = next.n; | |
nextNode = nextNode->next; | |
ListNode*& curList = lists[next.index]; | |
if(curList != NULL) { | |
node tmp; | |
tmp.n = curList; | |
tmp.index = next.index; | |
curList = curList->next; | |
heap.push_back(tmp); | |
push_heap(heap.begin(), heap.end()); | |
} | |
} | |
return result.next; | |
} | |
}; |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment