素数判断YES or NO

键盘输入一个正整数，要求判断该数是否为素数。素数即质数，只能被1和它本身整除。特别说明，1不是素数。如果是素数，输出“YES”，否则输出“NO”。

输入格式:

输入只有一个用例，即一个整数n。

输出格式:

如果n是素数，输出“YES”，否则输出“NO”。

#include<stdio.h>
#include<math.h>
int is_prime(int n)
{
 int i;
 float m;
 if(n==1)
  return 0;
 m=floor(sqrt(n)+0.5);
 for(i=2;i<=m;i++)
  if(n%i==0)
   return 0;
  return 1;
}
int main()
{
 int n,len;
 scanf("%d",&n);
    len=is_prime(n);
 if(len)
  printf("YES\n");
 else
  printf("NO\n");
 return 0;
}