这道题是那道 Merge two sorted lists的升级版,这次要求对多个链表进行排序。如果用链表来实现的话,可能会比较繁琐。这里我用了比较投机的一种方法,先用一个vector把所有链表的val存下来,再排序,最后再生成一个链表。这样写既简单,速度又快,不过可能偏离了这道题的本意了吧~
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
vector<int> values;
for (int i = 0; i < lists.size(); ++i)
{
ListNode * p = lists[i];
while(p != NULL){
values.push_back(p->val);
p = p->next;
}
}
sort(values.begin(), values.end());
ListNode * head = new ListNode(0);
ListNode * p = head;
for (int i = 0; i < values.size(); ++i)
{
ListNode * temp = new ListNode(values[i]);
p->next = temp;
p = p->next;
}
if(head == NULL)cout << 1;
head = head->next;
return head;
}
};
这里贴出答案里利用Merge two sorted lists来解决的版本,也很巧妙。
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.empty()){
return nullptr;
}
while(lists.size() > 1){
lists.push_back(mergeTwoLists(lists[0], lists[1]));
lists.erase(lists.begin());
lists.erase(lists.begin());
}
return lists.front();
}
ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) {
if(l1 == nullptr){
return l2;
}
if(l2 == nullptr){
return l1;
}
if(l1->val <= l2->val){
l1->next = mergeTwoLists(l1->next, l2);
return l1;
}
else{
l2->next = mergeTwoLists(l1, l2->next);
return l2;
}
}
};