Pregunta de entrevista de Citi

Coding challenge - writing a function