c语言检查素数简单程序

#include <iostream>
#include <stdio.h>
#include <math.h>
//简单素数的判断
using namespace std;

int PrimeNumber(int n){
int m,i;
m=sqrt(n);
if(n==2)
{return true;}
else if(n==1||n==0)
{return false;}
else
for(i=2;i<=m;i++)
{
if(n%i==0)
{return false;}
}
return true;
}

int main(){
int n,a;
while(scanf("%d",&n))
{
a=PrimeNumber(n);
if(a==0)
printf("%d isn't prime number \n",n);
else
printf("%d is prime number \n",n);
}

}