定义一个函数判断是否为素数（超简单！）

自定义函数：

bool prime(int a) {
    int b = 0;
    for(int i = 2; i 

接下来就是在主函数的if语句中调用该函数了

int main() {
    int num;

    cout << "input an integer: "<> num;
    cout<