LeetCode 203. 移除链表元素

题目描述

删除链表中等于给定值 val 的所有节点。

示例

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

解题思路

主要思路就是把当前节点的前继节点存其来，遇到和val相同的就可以给前继节点的next当前节点的next，这样就可以删除了，但是有连续的就有点麻烦了，所以用哨兵法来解决这个问题非常方便。

1. 先new一个哨兵初始化为0并让该哨兵的next指向head。
2. 当前节点为head，前继就是哨兵。
3. 直到当前节点不为空为止：
   (1) 如果当前节点的val等于要删除的值，那么该结点的前继就是该结点的next。要删除的节点存在toDelete里。
   (2)当前节点跳到下一个节点
   (3)如果toDelete不为空，那么删除toDelete，并初始化为nullptr。
4. 循环结束后返回哨兵的next即可。

代码

/**
 * 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* removeElements(ListNode* head, int val) {
        ListNode* sentinel = new ListNode(0);
        sentinel->next = head;
        ListNode* curr = head, *prev = sentinel, *toDelete = nullptr;

        while (curr != nullptr) {
            if (curr->val == val) {
                prev->next = curr->next;
                toDelete = curr;
            } else {
                prev = curr;
            }
            curr = curr->next;
            if (toDelete) {
                delete toDelete;
                toDelete = nullptr;
            }
        }
        return sentinel->next;
    }
};