有序链表的合并(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;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章