计蒜客 第三题 判断质数

对于大于 1 的数，如果除了 1 和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。

输入格式

输入包括一行，为一个整数 N(1

输出格式

输出包括一行，如果晓萌给出的整数 N 为质数，那么输出YES；如果 N 不是质数，那么输出NO。

样例输入

3
样例输出

YES

a = int(input())
b = 0
# 当 a等于1或2是 a为质数
if a == 1 or a == 2:
    print("YSE")

# 当a大于2是 用for循环遍历a与（2，a）的余数 ，当余为0时退出循环
else:
    for An in range(2,a):
        if a % An == 0:
            b = 1
            break
if b != 0:
    print("NO")
else:
    print("YES")