判断是否是素数

#include <stdio.h>
#include <math.h>
int prime(int n)
{
int i;
if(n==2||n==3) return 1;
if(n==1) return 0;
if(n%6!=1&&n%6!=5) return 0;
int M=sqrt(n);
for(i=5;i<=M;i+=6){
if(n%i==0||n%(i+2)==0) return 0;
}
return 1;
}
int main()
{
int n,p;
scanf("%d",&n);
p=prime(n);
printf("%s",p?"yes":"no");
return 0;
}