203. 移除链表元素

题目描述

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

示例:

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

分析

为了方便处理链表的首个节点，我们可以定义一个辅助的头结点dummy，然后让辅助结点指向链表表头，之后遍历链表，当碰到链表某个结点的值等于给定值，只需让当前结点的前一个结点指向该结点的后一个结点，同时删除该结点，最后返回dummy->next

具体代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        if(head == NULL) {
            return head;
        }
        ListNode *dummy = new ListNode(-1);
        dummy->next = head;
        ListNode* p = dummy;
        ListNode* s = NULL;
        while(p->next) {
            if(p->next->val == val) {
                s = p->next;
                p->next = p->next->next;
                delete s;
            } else {
                p = p->next;
            }
        }
        return dummy->next;
    }
};