leetcode203. Remove Linked List Elements

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

Given the head of a linked list and an integer val, remove all the nodes of the linked list that has Node.val == val, and return the new head.
套路 1

要想删除节点 node,必须在 node 的前一个节点执行删除操作。

例如链表 1→2→3,要想删除 2,必须在节点 1 处操作,也就是把节点 1 的 next 更新为节点 3。

套路 2

如果头节点可能被删除,那么要在头节点之前添加一个哨兵节点,这样我们无需特判头节点被删除的情况,从而简化代码逻辑。

或者说,根据套路 1,要想删除头节点,必须在头节点的前一个节点操作,所以要添加一个哨兵节点。

算法

复制代码
初始化哨兵节点 dummy,其 next 为 head。
遍历链表,初始化 cur=dummy。
循环直到 cur 的下一个节点为空。
如果 cur 的下一个节点的值等于 val,那么删除下一个节点,把 cur.next 更新为 cur.next.next。
如果 cur 的下一个节点的值不等于 val,那么不删除下一个节点,继续看下下一个节点是否要删除,即更新 cur 为 cur.next。
循环结束,返回 dummy.next,即删除节点后的新链表的头节点。

答疑

问:为什么没有修改 dummy,但 dummy.next 却是新链表的头节点?如果删除了 head,那么最后返回的是不是原链表的头节点?

答:注意初始化时,cur 和 dummy 都指向同一个节点,cur 和 dummy 只是同一个节点的引用,所以修改 cur.next 也会同时修改 dummy.next。

问:为什么删除下一个节点后,不需要更新 cur 为 cur.next?

答:删除下一个节点后,cur.next 的节点值也可能等于 val,也需要删除,如果直接更新 cur 为 cur.next,就漏删了节点。

python 复制代码
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        cur = dummy = ListNode(next=head)
        while cur.next:
            if cur.next.val == val:
                cur.next = cur.next.next  # 删除下一个节点
            else:
                cur = cur.next  # 继续向后遍历链表
        return dummy.next
相关推荐
萑澈1 小时前
算法竞赛入门:C++ STL核心用法与时空复杂度速查手册
数据结构·c++·算法·stl
Godspeed Zhao2 小时前
从零开始学AI16——SVM
算法·机器学习·支持向量机
江屿风2 小时前
C++OJ题经验总结(竞赛)1
开发语言·c++·笔记·算法
nebula-AI2 小时前
人工智能导论:模型与算法(核心技术)
人工智能·深度学习·神经网络·算法·机器学习·集成学习·sklearn
运筹vivo@2 小时前
LeetCode 2405. 子字符串的最优划分
c++·算法·leetcode·职场和发展·哈希表
数智工坊2 小时前
视觉-语言-动作模型解剖学:从模块、里程碑到核心挑战
论文阅读·人工智能·深度学习·算法·transformer
有点。3 小时前
C++(枚举法一练习题)
开发语言·c++·算法
黎阳之光3 小时前
视听融合新范式!黎阳之光打破视觉边界,声影协同赋能全域智慧管控
大数据·人工智能·物联网·算法·数字孪生
玖釉-3 小时前
栈——栈的定义及基本操作
c++·windows·算法·图形渲染
ゆづき4 小时前
Java 初学者入门指南:常见问题 + 核心知识点 + 进阶 20 道练习题
java·开发语言·学习·算法·水题