题目链接:203. 移除链表元素 - 力扣(LeetCode)
题目描述:
给你一个链表的头节点 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 输出:[]
解题思路:虚拟头节点
考虑到如果要移除的元素如果是头节点的话需要head = head.next,那么每次都需要判断,头节点和非头节点需要不同的处理方式较为复杂,所以此处引入虚拟头节点避免了这一问题。
移除元素时需要知道上一个元素,将上一个元素指向要移除元素的下一个元素,所以需要初始化一个指针,指向当前节点,遍历链表进行循环,循环条件是当前指针的下一个元素不为空。
代码:
class ListNode:def __init__(self,val = 0,next = None):self.val = valself.next = next
class Solution:def removeElements(self,head:Optional[ListNode],target:int)->Optional[ListNode]:# 初始化虚拟头节点dummy_head = ListNode(next = head)# 初始化指针cur = dummy_headwhile cur.next:if cur.next.val == targetcur.next = cur.next.nextelse:cur = cur.nextreturn dummy_head.next