Pregunta de entrevista de Yandex

two easy-medium leetcode problems