Nubank interview question

Resolver un algoritmo en python