203. 移除链表元素【链表】【C++】

news2024/11/16 6:26:41

题目描述

题目描述
给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回新的头节点。
示例 1：
在这里插入图片描述

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]
示例 2：
输入：head = [], val = 1
输出：[]
示例 3：
输入：head = [7,7,7,7], val = 7
输出：[]
提示：
列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

思路：

本题属于典型的链表删除题目，删除链表的核心代码为:

cur -> next = cur -> next -> next;

就是跳过下一个的节点，直接和下一个的下一个节点相连，但需要注意，C++需要手动释放内存。
为了方便处理第一个节点，可以设置一个虚拟头节点。

代码：

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* dummyHead = new ListNode(0);  //虚拟头节点，用于处理第一个节点
        dummyHead -> next = head;
        ListNode* cur = dummyHead;  //临时节点，用于遍历链表
        while (cur -> next != NULL) {  //终止条件
            if (cur -> next -> val == val) {   //删除节点
                ListNode* tmp = cur -> next; 
                cur -> next = cur -> next -> next;
                delete tmp;          //手动释放内存
            }else {
                cur = cur -> next;   //遍历下一个节点
            }
        }
        ListNode* res = dummyHead -> next;
        delete dummyHead;
        return res;
    }
};