leetcode——第203题——虚拟头结点

最新推荐文章于 2024-08-21 02:30:30 发布

原创最新推荐文章于 2024-08-21 02:30:30 发布 · 239 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #单链表 #c++

类型总结之——链表篇专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种从单链表中移除特定值的所有节点的方法。通过使用虚拟头节点简化边界条件处理，实现了高效的链表操作。文章提供了一个C++示例代码，展示了如何遍历链表并移除值等于给定值的节点。

题目：
给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回新的头节点。

/**
 * 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 dummy(0);
        dummy.next = head;
        ListNode* cur = &dummy; // 设置虚拟头结点

        while (cur->next)
        {
            if (cur->next->val == val)
            {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp; // 防止内存泄漏
            }
            // 这里的 else 是不能少的，因为如果少了的话，就是少判断一个
            else    cur = cur->next;
        }
        return dummy.next;
    }
};