用函数判断素数

#include<stdio.h>
 void main(){
  int prime(int n);
  int n;
  printf("\nInput an integer:\n");
  scanf("%d",&n);
  if(prime(n))
    printf("\n%d is a prime.\n",n);
    else
    printf("\n%d is  not a prime.\n",n);
 }
 int prime(int n)
 {
 int flag =1,i;
 for(i=2;i<=n/2&&flag;i++)
    if(n%i==0)flag=0;
    return(flag);
 }