函数判断素数

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

#include <stdio.h>
int main ()
{
    int sushu(int a);
    int b;
    printf ("输入一个数：");
    scanf ("%d",&b);
    sushu(b);
    return 0;
}

int sushu(int a)
{
    int i;
    for (i=2;i<=a;i++)
    {
        if(a % i == 0)
        break;
    }
    if (a == i)
        printf ("%d是素数\n",a);
    else
        printf ("%d 不是素数\n",a);
    return 0;
}