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

输入格式 

输入包括一行，为一个整数 N,N大于1小于1000； 

输出格式 

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

from math import sqrt
num = int(input())
if (0 not in [num % d for d in range(2,int(sqrt(num)+1))]):
    print("YES")
else:
    print("NO")