自定义函数:
bool prime(int a) {
int b = 0;
for(int i = 2; i <a; i++) { //i为除了1和它本身的数
if(a % i == 0) { //如果a能整除除了1和它本身的数
b++;
}
}
if(b == 0) { //b==0说明在上面的程序中,a没有整除任意一个i
return true; //b为素数
}
else {
return false;
}
}
接下来就是在主函数的if语句中调用该函数了
int main() {
int num;
cout << "input an integer: "<<endl;
cin >> num;
cout<<endl;
if (prime(num)) {
cout << num<<" is a prime." << endl;
} else {
cout << num<<" is not a prime." << endl;
}
return 0;
}