题解 | 判断质数

import math 

def is_prime(m):
    if m==1:
        return False 
    elif m==2:
        return True
    elif m%2==0:
        return False
    else:
        max_divisor=math.isqrt(m)+1
        for i in range(3,max_divisor):     
            if m%i == 0:
                return False
        return True

if is_prime(int(input())):
    print("Yes")
else: 
    print("No")