hackerrank sparse arrays problem solution in python