Python移除链表元素-虚拟节点
题目:
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
思路:
A.创建一个虚拟头部节点,这样能确保就算需要删除的数字为头部节点的值,也不用进行特殊处理。
B.创建一个current指针指向虚拟头部节点。
C.遍历链表,若当前current指针所指向节点的下一个节点数值为需要删除的数值,则把当前current指针的下一位指向下下个节点,直接跳过下一个节点。
如图:
D.若不是,则继续遍历剩下的链表节点。
E.遍历结束后,返回虚拟头部节点的下一位,即真实的头部节点。
代码:
class Solution:def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:# 创建虚拟头部节点dummy_head = ListNode(next = head)# 将current指针指向虚拟节点current = dummy_head# 当current指针遍历链表并删除值为val的节点while current.next:if current.next.val == val:current.next = current.next.nextelse:current = current.nextreturn dummy_head.next