Pergunta de entrevista da empresa Citadel

Solve problem in python please