写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。

//判断是不是素数 
#include
int prime(int a);//函数声明
int main()
{
int n,i;
scanf("%d",&n);
if(prime(n)==1) //return返回值1
printf("prime");
else
printf("not prime") ;
}
int prime(int a)
{
int i;
for(i=2;i if(a%i==0)
return 0;//a%i==0不成立
else 
return 1;//a%i==0成立
}