只是简单的判断素数。

#include<bits/stdc++.h>
using namespace std;
#define int long long
int a;
void zhi(){
    cin>>a;
    if(a<2){
        cout<<"not prime";
        return;
    }
    for(int i=2;i*i<=a;i++){
        if(!(a%i)){
            cout<<"not prime";
            return;
        }
    }
    cout<<"prime";
}
signed main(){
    zhi();//这一行可以替换成函数里的内容
    return 0;
}

C++ :

#include<iostream>
#include<cmath>
using namespace std;
bool prime(int n)
{
	int i,t=sqrt(n);
	for (i=2; i<=t; i++)
		if (n%i==0) break;
	return i>t;
}
int main()
{
	int n;
	cin>>n;
	if (prime(n)) cout<<"prime"<<endl;
	else cout<<"not prime"<<endl;
	return 0;
}