1012 Prime Number ?

#include<stdio.h>

#include<math.h>

int a,b,c,i,d;

int main()

{

 while(scanf("%d",&a)!=EOF){

  b=(int)sqrt((double)a);

  for(i=2;i<=b;i++){

   if(a%i==0)

   break;

  }

  if(i>b&&a!=1)

  printf("YES\n");

  else

  printf("NO\n");

 }

 return 0;

}