【c语言】判断一个数是不是素数？

#include<stdio.h>
#include<math.h>
int prime(int n){
    int i;
    if(n<2){
        printf("数据错误! ");
        return -1;
    }
    for(i=2;i<=sqrt(n);i++){
     if(n%i==0){
         printf("%d不是素数! ",n);
            return 1;
        }
    }
    if(i>=sqrt(n)){
    printf("%d是素数! ",n);
            return 0;
        }
}

void main(){
    int n;
    while(1){
    printf("请输入一个数字:");
    scanf("%d",&n);
    prime(n);
    
    }

}