簡單的兩個有序鏈表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;
}
};