Pergunta de entrevista da empresa Schrödinger

Started with DSA question on array