Pregunta de entrevista de Revolut

Python logic problem of live coding