Pregunta de entrevista de Citadel

2 hour assessment in python