文章目录
Leetcode 203: 移除链表元素 (python)
思路
这里采用常规的链表元素移除思路:构建一个虚拟头节点指向题目所给的头节点head,然后依次遍历链表中的元素,判断其值是否等于val,若相等,则去除当前元素;否则,将当前指向的元素后移一个,然后继续判断。
代码如下:
# 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:
# 这里head指向列表中的第一个ListNode
# 所以我们需要创建一个虚拟头节点
vHead = ListNode(0, head)
p = vHead
while p.next != None:
if p.next.val == val:
p.next = p.next.next
else:
p = p.next
return vHead.next