Здравствуйте, имею код, выдает ответ правильно, но не проходит по времени в тестирующей системе. Как мне его изменить, чтобы код стал работать быстрее?
Код:
def isPrime(n):
if n in [2,3]:
return("YES")
else:
if n%2==0:
return("NO")
else:
for i in range(3,n//2+1,2):
if n%i==0:
return("NO")
return("YES")
n = int(input())
print(isPrime(n))