Pregunta de entrevista de Revolut

How would you solve the following question using python etc