Pergunta de entrevista da empresa NDS

re writing the function logic