如何在函数开头加一个素数测试

int Pream(int n){/*参数n*/
int i,flag=0; /*初始标志为0*/
if(n==2)flag=1; /*n为2时，为素数，标志为1，跳过else，直接返回值1*/
else{
for(i=2;i<=n/2;i++){
if(n%i==0){flag=1;break;}} /*n不为2时，大于2时，循环是合数的有语句flag=1，返回1；小于2时，循环不执行，不是素数，返回0*/
}
return flag;/*返回*/
}