一个基本定理:如果一个数n是合数,那么它的所有因子都不超过sqrt(n)。我们可以利用这个方法判断是否是素数
import math
N = int(input())
if N == 1:
print(False)
else:
flag = True # 是质数
for i in range(2, int(math.sqrt(N)) + 1, 1):
if N % i == 0:
flag = False # 不是质数
break
if flag == True: #是质数
print("YES")
else:
print("NO")
近期评论