Pregunta de entrevista de Citadel

Solve problem in python please