21 合併兩個有序連結串列
阿新 • • 發佈:2018-12-24
合併兩個有序連結串列
普通方法
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; }