hackerrank day 2 basic probability problem solution in python