素数判断函数 c

#include<stdio.h>
int su(int a)
{
    int i;
    if(a==1||a==0)
        return 0;
    if(a==2||a==3)
        return 1;
    for(i=2; i*i<=a; i++)
    {
        if(a%i==0)
            return 0;
    }
    return 1;
}
int main()
{
    int a;
    scanf("%d",&a);
    if(su(a)==0)
    printf("not prime");
    else
        printf("prime");
    return 0;

}