leetcode-203.移除链表元素

题目描述

代码提交

代码

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode *dummyhead = new ListNode(0); // 设置一个虚拟头结点,堆上
        dummyhead->next = 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;
    }
};