leetcode 203. 移除链表元素

203. 移除链表元素

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

示例 1：

输入：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

输出：[]

提示：

• 列表中的节点数目在范围 [0, 104] 内
• 1 <= Node.val <= 50
• 0 <= val <= 50

题目链接： 203. 移除链表元素 - 力扣（LeetCode） (leetcode-cn.com)

解题思路

• 如何在单链表中删除目标节点
  • 将目标节点的前一个节点直接指向下一个节点
• 删除头节点和其他节点是不同的
  • 删除头节点只需要将头节点向后移动一位
  • 可以设置一个哑结点指向头节点，这样删除所有节点的操作都一样了
  • 即dummy = ListNode() dummy.next = head
• 最后返回的是 dummy.next

# 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:
        # 首先可以设置一个哑结点,哑结点指向头节点
        dummy = ListNode()
        dummy.next = head
        # 设置一个指针指向当前比较的节点值的前一个节点
        # 于是在开始阶段该指针指向哑结点
        p = dummy
        # 循环遍历整个链表
        while p:
            # 找到要删除的节点
            if p.next and p.next.val == val:
                # 删除该节点
                p.next = p.next.next
            # 该节点不是则跳过检查下一个
            else:
                p = p.next
            # 最后返回哑结点的下一个节点
        return dummy.next