C语言 判断一个数是不是素数（用函数表示）

# include <stdio.h>

int f(int n)  //定义函数f

{

    int i;
    int sum=0;
    if(n!=2)    
    {
        for(i=2;i<n;i++)
     {
        if(n%i==0)
           {
               sum=0;
               break;
           }
        if(n%i!=0)
            sum=1;
     }
    }
    if(n==2)    
        sum=1;
    return sum;
}

int main ()
{

    int n;
    scanf("%d",&n);
    if(f(n)==1)
    printf("yes");
    else
        printf("no");
    return 0;

}