Pergunta de entrevista da empresa MathWorks

Hackerrank - 1Medium, 1 Medium-Hard LC question