LeetCode 203.移除链表元素

LeetCode 203.移除链表元素
**题目描述：**给定链表头节点head和整数val，删除所有值为val的节点 ，返回新的头
视频讲解
文章讲解

• 思路：
  • 设置虚拟头节点，统一删除head和非head节点的操作
  • cur从虚拟头节点开始遍历，寻找待删节点的前驱，即判断cur->next->val与val
  • 若相等则删除，注意释放空间；若不等，cur向后走一位
  • 注意不能直接返回head（可能被删掉）

/**
* 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* dummyHead = new ListNode(0, head);
    // 从虚拟头节点开始遍历
    ListNode* cur = dummyHead;
    while (cur->next != nullptr) {
        if (cur->next->val == val) {
            ListNode* tmp = cur->next;
            cur->next = cur->next->next;
            delete tmp;
        } else {
            cur = cur->next;
        }
    }
    head = dummyHead->next;
    delete dummyHead;	//释放虚拟头节点
    return head;
}
};