删除列表元素
之前出问题的地方是没考虑到空链表
# 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]:
while head and head.val==val:
head = head.next
if head:
pre = head
node = pre.next
while node:
if node.val == val:
pre.next = node.next
else:
pre = pre.next
node = node.next
return head
运行结果: