编写一个判断素数c语言函数

以下是一个判断素数的C语言函数的例子：

int prime(int a) {
    int i;
    for(i=2;i<a;i++) {
        if(a%i==0) {
            return 0;
        }
    }
    return 1;
}

在主函数中输入一个整数，调用prime判断，然后在主函数中输出判断结果：

#include <stdio.h>

int prime(int a);

int main() {
    int n,k;
    scanf("%d",&n);
    k=prime(n);
    if(k==1) {
        printf("yes\n");
    } else {
        printf("no\n");
    }
    return 0;
}