LeetCode 19 Two Pass attempt 1

19_remove_nth_node_from_end_of_list.py

from typing import Optional


class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next


class Solution:
    def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
        """Remove the nth from the last node.

        Args:
            head (Optional[ListNode]): Head of list
            n (int): Node to remove

        Returns:
            Optional[ListNode]: Modified list
        """
        length = 0
        current = head
        while current:
            current = current.next
            length += 1

        node_to_remove = length - n
        current = head
        for _ in range(node_to_remove - 1):
            current = current.next
        if current.next:
            current.next = current.next.next
        else:
            current.next = None

        return head