dynamic array hackerrank solution in python