1. 程式人生 > 其它 >[題解]劍指 Offer 18. 刪除連結串列的節點(C++)

[題解]劍指 Offer 18. 刪除連結串列的節點(C++)

題目

給定單向連結串列的頭指標和一個要刪除的節點的值,定義一個函式刪除該節點。

返回刪除後的連結串列的頭節點。

注意:此題對比原題有改動

示例 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;
    }
};