方法一
创建一个链表res,依次和lists里面的链表,进行两个链表的合并操作。比如,res初始化为空,先和lists[0],合并获得新的res;res再和lists[1],合并又获得新的res…
最终就得到的有序的链表。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
//辅助函数功能:合并两个升序链表
ListNode* mergeTwoLists(ListNode* l1,ListNode* l2){
if(!l1&&l2)return l2;
if(!l2&&l1)return l1;
ListNode *head=new ListNode();
ListNode *node=head;
while(l1&&l2){
if(l1->val<=l2->val){
node->next=l1;
node=node->next;
l1=l1->next;
}else{
node->next=l2;
node=node->next;
l2=l2->next;
}
}
if(l1) node->next=l1;
if(l2) node->next=l2;
return head->next;
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
//一定要初始化,否则报错
ListNode *res=nullptr;
int n=lists.size();
for(int i=0;i<n;++i){
res=mergeTwoLists(res,lists[i]);
}
return res;
}
};
方法二:
分治的方法,来做,先分成一个一个的单链表,然后两两合并,最后就得到排好序的最终链表。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1,ListNode* l2){
if(!l1&&l2)return l2;
if(!l2&&l1)return l1;
ListNode *head=new ListNode();
ListNode *node=head;
while(l1&&l2){
if(l1->val<=l2->val){
node->next=l1;
node=node->next;
l1=l1->next;
}else{
node->next=l2;
node=node->next;
l2=l2->next;
}
}
if(l1) node->next=l1;
if(l2) node->next=l2;
return head->next;
}
ListNode* merge(vector<ListNode*>&lists,int l,int r){
if(l==r)return lists[l];
if(l>r)return nullptr;
int mid=(l+r)/2;
return mergeTwoLists(merge(lists,l,mid),merge(lists,mid+1,r));
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
return merge(lists,0,lists.size()-1);
}
};