Pregunta de entrevista de Apple

Leetcode q - forgot the exact one