计蒜客 【判断质数】

【判断质数】

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

输入格式

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

输出格式

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

Java

import java.util.Scanner;

public class Main {
    public static boolean judgeNum(int n) {
        if(n < 2) return false;
        if(n == 2) return true;
        if(n > 2){
            for(int i = 2;i <=n/2;i++){
                //i<=(int)Math.sqrt(n)
                if(n % i == 0) return false;
            }
        }
        return true;
    }
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        System.out.print(judgeNum(n)?"YES":"NO");
    }
}

Python

n = int(input())
if n > 1:
    for i in range(2,n):
        if (n % i) == 0:
            print("NO")
            break
    else:
        print("YES")
else:
    print("NO")
//参考：http://www.runoob.com/python3/python3-prime-number.html