1. 程式人生 > >21 合併兩個有序連結串列

21 合併兩個有序連結串列

合併兩個有序連結串列

普通方法

ListNode *mergeTwoLists(ListNode *&l1, ListNode *&l2) {
        ListNode head(0), *temp = &head;
        while (l1 != nullptr && l2 != nullptr) {
            if (l1->val < l2->val) {
                temp->next = l1;
                l1 = l1->next;
            } else {
                temp->next = l2;
                l2 = l2->next;
            }
            temp = temp->next;
        }
        if (l1 != nullptr)
            temp->next = l1;
        else if (l2 != nullptr)
            temp->next = l2;
        return head.next;
    }

考慮這樣一種情況,l1連結串列內容為(1,2,3,4,5),l2的內容為(6,7,8,9),在上述程式碼中,即使前五個數字都來自於l1,但是程式碼仍將執行 temp->next=l1 五次,浪費時間,可進一步優化

最優解

    ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) {
        ListNode dummy(0);
        dummy.next = l1;
        ListNode *cur = &dummy;

        while (l2) {
            while (cur->next && cur->next->val <= l2->val)
                cur = cur->next;
            l1 = cur->next; 
            cur->next = l2;
            l2 = l1;
        }
        return dummy.next;
    }