Pregunta de entrevista de NDS

re writing the function logic