python移除链表元素

｜ 移除链表元素

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

输入：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
输出：[]

｜ 题解

# 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: ListNode, val: int) -> ListNode:
        # 解决链表常见的三种情况
        # 1.头节点是否需要特殊处理
        # 2.尾节点是否需要特殊处理
        # 3.空链表是否需要特殊处理
        
        # 常见的处理方式 引入虚拟头节点
        # 引入虚拟头节点
        new_head = ListNode
        new_head.next = head
        prev = new_head
        while prev.next is not None:
            # 如果要删除某个节点，首先要找到被删除节点的前驱节点
            if prev.next.val == val:
                prev.next = prev.next.next
            else:
                prev = prev.next
        return new_head.next