计蒜客 判断质数 Java实现

import java.util.Scanner;
public class Main{
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
int a=sc.nextInt();
if(happy(a)){
System.out.println("NO");
}
else{
System.out.println("YES");
}
}
public static boolean happy(int a){
if(a==0||a==1){
return false;
}
for(int i=2;i<=Math.sqrt(a);i++){
if(a%i==0){
return true;
}
}
return false;
}
}