Pergunta de entrevista da empresa JPMorganChase

Probability question, python, easy coding