编写函数int prime(int n)判断n是否为素数

int prime(int n) {
if (n <= 1) { // 小于等于1的数不是素数
return 0;
}
for (int i = 2; i <= sqrt(n); i++) { // 从2到n的平方根遍历
if (n % i == 0) { // 如果n能被i整除，说明不是素数
return 0;
}
}
return 1; // 否则是素数
}