Ответ:

import sys

sys.setrecursionlimit(22000)

def func(a, i):

 if i*i<a:

   if a>=i:

     if a != i and a%i != 0:

       i+=1

       return(func(a, i))

     elif a == i:

       return ('YES')

     else:

       return('NO')

   else:

     return('NO')

 else:

   return('YES')

i = 2

a = int(input())

print(func(a,i))

вот, первые две строчки можете убрать, если на проверки вводятся не слишком большие числа.

буду благодарен, если отметите лучшим решением:)

Объяснение: