計蒜客題庫（Python）：三、判斷質數

對於大於1的數，若是除了1和它自己，它不能再被其它正整數整除，那麼咱們說它是一個質數。曉萌想判斷一個數是否是質數，但願找你寫個程序，幫助她進行判斷。

輸入包括一行，爲一個整數N（1 < N ≤1000），正是曉萌給出你讓你判斷的數字。

輸出包括一行，若是曉萌給出的整數N爲質數，那麼輸出YES；若是N不是質數，那麼輸出NO。

樣例1

輸入：

3

輸出：

YES

題解：判斷一個數是否爲質數是看它可否被除了1和自己之外的整數整除。一般用循環從1開始直到N-1進行整除判斷，有一個被整除就不是質數，直接返回NO。由數學知識可知，判斷1到√N便可。

N=int(raw_input(""))
def isPrime1(n):
    for i in range(2, int(n**0.5)):#使用循環判斷N可否被從2到√N的數整除
        if n % i == 0:
            return "NO"
    return "YES"
print isPrime1(N)