判断质数

#include<stdio.h>
#include<math.h>
int p(int x)
{
int k,i,leap=1;
k=(int)sqrt(x);
for (i=2;i<=k;i++)
{
if (x%i==0)
{
leap=0;
return 0;
break;
}
}
if (leap==1)
return 1;
}
int main()
{
int x,i;
while (scanf("%d",&x)!=EOF)
{
i=p(x);
if(i==0)
printf("No\n");
else if (i==1)
printf("Yes\n");
}
}