LeetCode(2)——链表——移除链表元素

203. 移除链表元素

题目大意：
给定一个链表和一个target，将链表总节点值等于target的节点删除。

思路：
因为头节点可能被删除，所以为了让操作统一，我们加一个虚拟头节点。

题解：

/**
 * 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) {
        if(head == nullptr)
            return head;
        ListNode* p = new ListNode(0, head);
        //p->next = head;
        ListNode* res = p;
        while(p->next != nullptr)
        {
            if(p->next->val == val)
            {
                p->next = p->next->next;
            }
            else
                p = p->next;
        }
        return res->next;
    }
};