Pergunta de entrevista da empresa ASM

Leet code easy, an array question.