Pergunta de entrevista da empresa DeepX (Japan)

On second interview, writing a function that takes a number as an input and returns if it is a prime or not