C语言函数实现判断一个数是不是素数

#include <stdio.h>
#include <math.h>
#define yes 1
#define no 0
void main()
{
 int n;
 int prime_num(int n);//声明prime_num函数
  
 printf("please enter a number:");
 scanf("%d",&n);//输入要判断的的数
 prime_num(n);//调用prime_num函数
 if(prime_num(n)==yes)
  printf("%d is a prime number\n",n);
 else
  printf("%d is not a prime number\n",n);
}
int prime_num(int n)//定义prime_num函数
{
 int i;
 if(n==2)
  return yes;
 else
  if(n<2||n%2==0)
   return no;
  else
  {
   for(i=3;i<=sqrt(n);i+=2)
   {
    if(n%i==0)
     return no;
   }
  }
  return yes;
}