输入一个数，判断其是否为素数。。1.我想用（int）sqrt(n),可是加上后就运行不了2.有规定for里面不能加if吗？不能嵌套？还是怎样？老师说没说我也忘记了3.最最重要的能不能把printf放在后面的int main（）里？？？我这样写总感觉跟没用函数一样。。。
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
int judge()
{
    int n,i,k,flag;
    printf("请输入这个数：");
    scanf("%d",&n);
        if(n==0||n==1)
            printf("不属于质数讨论范围");
        else if (n==2)
            printf("这个数是素数");
        flag=1;
        for(i=2;i<n;i++)
    {
        if(n%i==0)
            printf("这个数不是素数");
        else
            printf("这个数是素数");   
        return flag;
   
    }
}
    int main()
    {
        judge();

        system("pause");
        return 0;
    }