Pergunta de entrevista da empresa Citi

Coding challenge - writing a function