zcmu1012：找素数（时间效率优化）

Description

请判断给出的n是不是素数？

Input

输入每一行包含一个正整数n，n<=231-1;

#include<stdio.h>
#include<math.h>
int main()
{  
long long n;
while(scanf("%lld",&n)!=EOF)
{   int ifprime=0,i;
   if(n==1)
   {
   	ifprime=1;
 
   }
	for(i=2;i<=sqrt(n);i++)
	{
		if(n%i==0)
		{
			ifprime=1;//是素数 
			break;
		}
	}
	if(!ifprime)
	{
		printf("YES\n");
	}else
	{
		printf("NO\n");
	}
}
}

Output

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

Sample Input

4 5 6

Sample Output

NO YES NO