函数——判断素数

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

#include <stdio.h>
int primer_number(int n)
{
    int a, b;
    int flag;
    flag=1;
    for (a=2; a<=n/2; a++)
    {
        if (n%a==0)
        {
            flag=0;
            continue;
        }
    }
    if (flag==1)
    {
        b=1;
    }
    return b;
}
int main()
{
    int num, c;
    scanf("%d", &num);
    c=primer_number(num);
    if (c==1)
    {
        printf("为质数\n");
    }
    else
    {
        printf("为合数\n");
    }
    return 0;
}