Pregunta de entrevista de Trexquant Investment

Asked a leetcode medium level question on arrays. Had to solve in python.