写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息

小白在坚持刷题······
题目描述
写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。
输入
一个数
输出
如果是素数输出prime 如果不是输出not prime

#include 
using namespace std;
bool isprime(int n)
{
    if(n > 0 && n < 4){return true;}
    for(int i = 4; i < n; i++)
    {
        if(n % i == 0){return false;}
    }
    return true;
}
int main()
{
    int n;
    cin >> n;
    if(isprime(n)){cout << "prime" << endl;}
    else {cout << "not prime" << endl;}
    return 0;
}