对于大于 1 的数，如果除了 1 和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。
输入格式
输入包括一行，为一个整数 N(1<N≤1000)，正是晓萌给出你让你判断的数字。
输出格式
输出包括一行，如果晓萌给出的整数 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")