判断一个数是不是素数

#include<stdio.h>
#include<math.h>
int main()
{
	int n,i;
	while(scanf("%d",&n)!=EOF)
	{
		int flag=1;
		for(i=2;i<=sqrt(n)&&flag;i++)
		{
			if(n%i!=0)
			{
				flag=1;
			}
			else
				flag=0;
		}
		if(flag==0||n==1)
		{
			printf("不是素数\n");
		}
		else
		{
			printf("是素数\n");
		}
	}


}