判断素数[函数]，请按题解答

Question

题目描述
写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。
输入格式
一个数
输出格式
如果是素数输出prime 如果不是输出not prime
样例输入
97
样例输出
prime

Answer 1

以下是一个简单的C++程序，包含一个判断素数的函数和主函数：

#include <iostream>
#include <cmath>

// 判断素数的函数
bool isPrime(int num) {
    if (num <= 1) {
        return false;  // 1不是素数
    }

    int sqrtNum = sqrt(num);
    for (int i = 2; i <= sqrtNum; ++i) {
        if (num % i == 0) {
            return false;  // 能被整除，不是素数
        }
    }

    return true;  // 未被整除，是素数
}

int main() {
    int num;

    // 输入一个整数
    std::cout << "输入一个整数: ";
    std::cin >> num;

    // 判断并输出是否是素数
    if (isPrime(num)) {
        std::cout << "prime" << std::endl;
    } else {
        std::cout << "not prime" << std::endl;
    }

    return 0;
}