hackerrank day 3 drawing marbles problem solution in python