Pregunta de entrevista de Firework

Python Algorithmic Questions, nothing super tricky