Pregunta de entrevista de Citi

Basic array manipulation and matrix operation for coding