Domanda di colloquio di Citadel

Solve problem in python please