给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
示例:
输入:head = [1,2,6,3,4,5,6], val = 6 输出:[1,2,3,4,5]
解法一:虚拟头节点
考虑到链表没有指针指向头节点,所以删除头节点时需要特殊的逻辑,所以我们虚拟一个头节点指向原头节点将删除逻辑统一,最后return头节点之后的链表。
判断当前节点值是否是目标值,是的话断开链接,将节点指针指向下下个节点,不是的话移动到下一个节点继续判断。
# 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: Optional[ListNode], val: int) -> Optional[ListNode]:
dhead = ListNode(next=head)
current = dhead
while current.next:
if current.next.val == val:
current.next = current.next.next
else:
current = current.next
return dhead.next