thiagowfx's avatar

¬ just serendipity 🍀 (not just serendipity)

LeetCode #203: Remove Linked List Elements

• 67 words • 1 min • updated

LeetCode #203: Remove Linked List Elements:

python
# 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

        pointer = head
        while pointer and pointer.next:
            if pointer.next.val == val:
                pointer.next = pointer.next.next
            else:
                pointer = pointer.next

        return head