计蒜客 第3题:判断质数

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

输入包括一行，为一个整数N（1 < N ≤1000），正是晓萌给出你让你判断的数字。<n≤1000）。< style="box-sizing: border-box;">

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

样例1

输入：

3

输出：

YES

# coding=utf-8def sushufou(num):    if num in [0,1]:        #print('0和1不是素数')        return 'NO'    if num==2:        #print('是素数')        return 'YES'    for i in range(2,num):        if i*i >num:            #print( '是素数')            return 'YES'        if num%i==0:            #print( '不是素数,存在因数：',i)            return 'NO'                        num = int(raw_input(''))print sushufou(num)