题解 | 判断质数

import math
import sys

n = int(input())
if n == 1:
    print("No")
elif n == 2 or n == 3:
    print("Yes")
elif n % 2 == 0:
    print("No")
else:
    for i in range(3, int(math.sqrt(n)) + 1, 2):
        if n % i== 0:
            print("No")
            sys.exit()
    print("Yes")