LeetCode203.移除列表元素

题目

给你一个链表的头节点 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

代码

# 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:
        if head==None:
            return head
        while head!=None and head.val==val:
       	    head=head.next
        if head==None:
       	    return head
        p=head
        q=p.next
        while q!=None:
       	    if q.val==val:
       		    while q!=None and q.val==val:
       			    q=q.next
       		    p.next=q
       	    else:
       		    p=p.next
       		    q=q.next
        return head