判断素数 -- C++

#include <iostream>
using namespace std;

bool IsPrime(int n)
{
    if (n == 1 || n == 0) return false;
    if (n == 2) return true;
    for (int i = 2; i * i <= n; i ++ )
        if (n % i == 0) return false;
    return true;
}

int main()
{
    int flag, n;
    cin >> n;
    flag = IsPrime(n);
    cout << flag << endl;
    return 0;
}