判断是否为素数

#include
#include<cstdio>

using namespace std;

bool prime(int num);  //判断是否为素数。素数:大于1的自然数，且除了1和他本身以外没有其他正因数。

int main() {

    int num;
    while (cin >> num) {
        if (prime(num)) {
            printf("True\n");
        } else {
            printf("False\n");
        }
    }

    return 0;
}

bool prime(int num) {
    if (num < 2) {
        return false;
    } else {
        int count = 0;  //计数器
        for (int i = 2; i < num; ++i) { 
            if (!(num % i)) {  //判断除1和它本身以外还有没有正因数
                ++count;
            }
        }
        if (count == 0)
            return true;
        else
            return false;
    }
}