題目 21:「合併兩個有序鏈結串列 (Merge Two Sorted Lists)」 是一道經典的鏈結串列操作問題。目標是將兩個已經排序好的單向鏈結串列合併成一個新的排序鏈結串列。
題目:
給定兩個排序的單向鏈結串列 l1
和 l2
,將它們合併成一個新的排序鏈結串列。新鏈結串列是由這兩個鏈結串列的所有節點組成,並且也是有序的。
範例:
輸入:l1 = [1,2,4]
,l2 = [1,3,4]
[1,1,2,3,4,4]
輸入:l1 = []
,l2 = []
[]
輸入:l1 = []
,l2 = [0]
[0]
從list1與list2中選擇一個較小的作為root起始,curr作為root list上當前的節點,然後每次都比較當前的list1-val與list2-val誰比較小,較小者就接在curr之後,這樣結果就會是由小到大排列的list,
一開始可以先處理一些邊界情況,
實作:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
if (list1 == nullptr && list2 == nullptr) {
return nullptr;
}
if (list1 == nullptr) {
return list2;
}
if (list2 == nullptr) {
return list1;
}
ListNode* curr;
if (list1->val < list2->val) {
curr = list1;
list1 = list1->next;
} else {
curr = list2;
list2 = list2->next;
}
ListNode* root = curr;
while (list2 != nullptr && list1 != nullptr) {
if (list1->val < list2->val) {
curr->next = list1;
list1 = list1->next;
} else {
curr->next = list2;
list2 = list2->next;
}
curr = curr->next;
}
// list1 is nullptr or list2 is nullptr
// 當其中一個鏈結串列遍歷完,直接接上剩下的另一個鏈結串列
if (list1 == nullptr) {
curr->next = list2;
} else {
curr->next = list1;
}
return root;
}
};
參考:
#21. Merge Two Sorted Lists