判断质数

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

输入格式

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

输出格式

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

样例输入

3

样例输出

YES

import java.util.Scanner;


public class Main {

    public static void main(String[] args) {
        Scanner input= new Scanner(System.in);
        int num = input.nextInt();
        boolean ff = isPrim(num);
        if (ff) {
            System.out.println("YES");
        }else {
            System.out.println("NO");
        }
        
        

    }
    
    public static  boolean isPrim(int number){
        
        boolean flag = true;
        if (number == 1) {
            return flag;
        }
        for (int i = 2;i<=Math.sqrt(number);i++) {
            if (number %i == 0) {
                flag = false;
            }
        }
        return flag;
        
    }

}