diff --git a/Q2/solution.py b/Q2/solution.py index dd114fb..d09b857 100644 --- a/Q2/solution.py +++ b/Q2/solution.py @@ -6,3 +6,30 @@ class Solution: def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]: + prev_node = head + next_node = head + + count = 0 + + while prev_node.next != None: + + count += 1 + + if prev_node.next.value == value: + next_node = next_node.next.next + break + + prev_node = prev_node.next + next_node = next_node.next + + if head.value == value: + node = head + head = head.next + del node + + else: + del_node = prev_node.next + next_node = prev_node.next.next + del del_node + + return head