用c语言编写一个函数prime，判断一个正整数是否是素数，如果是素数，函数返回1，否则返回0

#include <stdio.h>

int prime(int n) {
    if (n <= 1) { // 1不是素数
        return 0;
    }
    for (int i = 2; i * i <= n; i++) { // 从2到根号n遍历
        if (n % i == 0) { // 如果n可以被i整除，说明n不是素数
            return 0;
        }
    }
    return 1; // n是素数
}

int main() {
    int n;
    printf("请输入一个正整数：");
    scanf("%d", &n);
    if (prime(n)) {
        printf("%d是素数\n", n);
    } else {
        printf("%d不是素数\n", n);
    }
    return 0;
}