Dau coroană! Cum scriu primalitatea în C++? Vreau scris in C++ ce am in imagine.

#include <iostream>
using namespace std;
int n,prim,i;
int main() {
cin>>n;
prim=1;
if(n==1 || n==0) prim=0;
else if(n%2==0 && n!=2) prim=0;
else for(i=3;i*i<=n;i+=2) if(n%i==0) prim=0;
if(prim==1) cout<<"Este prim";
else cout<<"Nu este prim";
}