简单的两个有序链表A 和 B 的合并,基本思路是每次从A[i] 和B[j] 中找到一个较大的,将其加入已经合并的新的链表的末尾,
C++ 代码:
ListNode * mergeTwoSortedList(ListNode* l1,ListNode *l2){
// time complexity O(m+n), space complexity O(1)
if(l1==NULL ) return l2;
if(l2==NULL) return l1;
//**生成一个表头,避免通过判断设置初始条件,使得代码精简、优雅。**
ListNode dummy(-1);
ListNode * p=&dummy;
while(l1 && l2){
if(l1->val<l2->val){
p->next=l1;
l1=l1->next;
}else{
p->next=l2;
l2=l2->next;
}
p=p->next;
}
p->next=l1?l1:l2;
return dummy.next;
}
如何合并k 个有序链表呢? 当然是利用已有的操作,每次将一个链表合并到已经合并的链表中即可。此过程时间复杂度为
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.size()==0) return NULL;
ListNode * p;
p=lists[0];
for(int i=1;i<lists.size();++i){
p=mergeTwoSortedList(p,lists[i]);
}
return p;
}
ListNode * mergeTwoSortedList(ListNode* l1,ListNode *l2){
if(l1==NULL ) return l2;
if(l2==NULL) return l1;
ListNode dummy(-1);
ListNode * p=&dummy;
while(l1 && l2){
if(l1->val<l2->val){
p->next=l1;
l1=l1->next;
}else{
p->next=l2;
l2=l2->next;
}
p=p->next;
}
p->next=l1?l1:l2;
return dummy.next;
}
};