C语言判断一个数是不是素数

#include <stdio.h>
#include <math.h>
//两种方法

int fun(int n)
/*{
    int flag=0;//假设是非素数
    int i;
    for(i=2;i<n;i++)//素数从2开始
    {
        if(n%i==0)
            break;//跳出for循环
    }
    if(i==n)//判断i是不是等于n
        flag=1;
    return flag;
}*/
{
    int flag=0,k=sqrt((float)n);//sqrt需要定义浮点型，k是int类型，会转化sqrt
    int i;
    for(i=1;i<k;i++)
    {
        if(n%i==0)
            break;//同上原理一致
    }
    if(i==k)
        flag=1;
    return flag;
}
void main()
{
    int n;
    printf("输入一个数\n");
    scanf("%d",&n);//定义输入变量
    if(fun(n))//通过输出1和0判断是不是素数