写一个判断素数的函数 在主函数输入一个整数 输出是否为素数的信息

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

#include<stdio.h>#include<math.h>int a;int prime(int a);int main(){int a;printf("Input a:\n");scanf("%d",&a);if(prime(a)){printf("%d是个素数\n",a);}elseprintf("%d不是素数\n",a);return 0;}int prime(int a){int flag=1,i;for(i=2;i<=sqrt(a)&& flag==1;i++){if(a%i==0){flag=0;}}return flag;}