Pregunta de entrevista de Amazon

Technical leet code medium question.