delete_index_ll.py

"""
Hackerrank solution.

Delete a node given a position.
"""

# 0 1 2 3 4
# 0 1   3 4
#   1 2 3 4

def Delete(head, position):
    
    curr = head
    prev = None
    
    if position == 0:
        head = curr.next
        return head
    
    else:
        while position > 0:
            prev = curr
            curr = curr.next 
            position -= 1
        
        prev.next = curr.next
        
        return head