203. 移除链表元素

from typing import Optional


# 定义 ListNode 类,这个要复杂粘贴
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]:
        # 创建一个虚拟头节点，简化删除操作
        dummy = ListNode(0)
        dummy.next = head
        p = dummy  # p 从虚拟头节点开始
#这里其实相当于创建了一个带头结点的链表，如果不这么做的话，删除元素还需要判断首节点元素
#是否和目标元素一样，比较复杂
        while p.next:  # 遍历链表
            if p.next.val == val:  # 如果找到匹配的节点，删除
                p.next = p.next.next
            else:  # 向后移动
                p = p.next

        return dummy.next  # 返回删除后的链表（虚拟头节点的 next）
# 辅助函数：将普通 Python 列表转换为链表，这个复制粘贴
def list_to_linkedlist(lst):
    if not lst:  # 如果输入的列表为空，返回 None
        return None
    head = ListNode(lst[0])  # 创建头节点
    current = head  # current 用于遍历链表
    for value in lst[1:]:  # 从第二个元素开始遍历
        current.next = ListNode(value)  # 创建下一个节点
        current = current.next  # 移动到下一个节点
    return head

# 辅助函数：打印链表，这个复杂粘贴
def print_linkedlist(head):
    current = head
    while current:
        print(current.val, end=" -> " if current.next else "")
        current = current.next
    print()
    # 测试
test = Solution()
head = list_to_linkedlist([1, 2, 6, 3, 4, 5, 6])  # 将普通列表转换为链表
print_linkedlist(test.removeElements(head, 6))  # 打印删除后的链表