Citi Interview Question

Basic array manipulation and matrix operation for coding