Pregunta de entrevista de MRSOOL

Simple/medium problem from leet code