判别素数的功能函数

编写一个判别素数的功能函数： int PrimeNum(int x)。要求：主函数中通过键盘输入一个整数后，调用PrimeNum函数返回1表示x是素数，返回0表示x不是素数。

#include <stdio.h>
int PrimeNum(int x);

int main()
{
    int i;
    while(1)
    {
        printf("请输入一个整数：");
        scanf("%d",&i);
            printf("%d\n",PrimeNum(i));
        continue;
    }
}

int PrimeNum(int x)
{
    int a,b,sum=0;
    for(a=2;a<x;a++)
    {
        if(x%a==0)
            sum=sum+1;
    }
    if(sum!=0||x==1||x==0)
    {
        b=0;
    }
    else
    {
        b=1;
    }
    return b;
}