Pregunta de entrevista de Amazon

Solve Leetcode question (leetcode medium)