Pergunta de entrevista da empresa Morgan Stanley

solve this problem in python