计蒜客刷题之4：判断质数 Python

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


输入格式
输入包括一行，为一个整数 N(1<N≤1000)，正是晓萌给出你让你判断的数字。


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

n = int(input())
val = 0

for i in range(2,n):
    if n % i == 0:
        val = 1
        print('NO')
        break

if val == 0:
    print('YES')