Leetcode203移除链表

题目描述

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

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        # 创建一个虚拟头节点，它的next指向原链表的头节点
        dummy = ListNode(0)
        dummy.next = head

        # 使用current指针遍历链接
        current = dummy
        while current.next:
            if current.next.val ==  val:
                current.next = current.next.next
            else:
                current = current.next
        return dummy.next

拓展总结

1. 使用虚拟头节点是链表操作中的常见技巧，它可以简化对头节点的删除操作。
2. 通过 while current.next 遍历链表，避免直接操作头节点，确保删除操作的灵活性。