有序鏈表的合併(in C++)

簡單的兩個有序鏈表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 個有序鏈表呢? 當然是利用已有的操作,每次將一個鏈表合併到已經合併的鏈表中即可。此過程時間複雜度爲 O(n1+n2+...+nk) ,空間複雜度O(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;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章