Java - 计蒜客 - 判断质数

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

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

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

样例输入

3

样例输出

YES

import java.util.Scanner;
public class Main{
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
int a = sc.nextInt();

if(isPrimeNum(a)){
System.out.println("YES");
}else{
System.out.println("NO");
}
}
public static boolean isPrimeNum(int a){
if(a==1||a==0){
return false;
}
for(int i=2 ;i<=Math.sqrt(a) ;i++){
if(a%i==0){
return false;
}
}
return true;
}
}