用C词语言程序写判断一个数是不是素数

#include <stdio.h>
#include <math.h>

int prime(int n)
{
int i,k;
k=sqrt(n);
for(i=2;i<=k;i++)
{
if(n%i==0) break;
}
if(i>k)
return 1;
return 0;
}

void main()
{
int n;
scanf("%d",&n); //从键盘输入一个整数.判断这个数是不是素数.
if(prime(n))
printf("%d is a prime number\n",n);
else
printf("%d is not a prime number\n",n);
}