判斷是否爲素數

#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;
    }
}