[題解]劍指 Offer 18. 刪除連結串列的節點(C++)
阿新 • • 發佈:2021-08-21
題目
給定單向連結串列的頭指標和一個要刪除的節點的值,定義一個函式刪除該節點。
返回刪除後的連結串列的頭節點。
注意:此題對比原題有改動
示例 1:
輸入: head = [4,5,1,9], val = 5
輸出: [4,1,9]
解釋: 給定你連結串列中值為5的第二個節點,那麼在呼叫了你的函式之後,該連結串列應變為 4 -> 1 -> 9.
示例 2:
輸入: head = [4,5,1,9], val = 1
輸出: [4,5,9]
解釋: 給定你連結串列中值為1的第三個節點,那麼在呼叫了你的函式之後,該連結串列應變為 4 -> 5 -> 9.
說明:
·題目保證連結串列中節點的值互不相同
·若使用 C 或 C++ 語言,你不需要 free 或 delete 被刪除的節點
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/shan-chu-lian-biao-de-jie-dian-lcof
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
思路
基本思想就是遍歷連結串列,找到要刪除的節點把它刪掉就行了。怎麼刪除一個節點?只需要讓該節點前面的節點的next指標指向該節點的下一個節點,所以我們要找到需要刪除的節點的前一個節點。注意,如果要刪除的是頭節點,沒有前面的節點該怎麼辦?很簡單,可以在連結串列頭新增一個啞節點dummyHead。
時間複雜度是O(n),空間複雜度O(1)。
程式碼
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* deleteNode(ListNode* head, int val) { ListNode *dummyHead = new ListNode(0); dummyHead->next = head; ListNode *cur = dummyHead; while(cur->next && cur->next->val != val) { cur = cur->next; } if(cur->next == nullptr) return dummyHead->next; cur->next = cur->next->next; return dummyHead->next; } };