Pregunta de entrevista de Amazon

Find an alternative way to optimize the algorithm.