i_prim prima varianta

#include <iostream>
#include <cmath>

using namespace std;
int prim (int x)
{
    int i,ok=0;
    for(i=2;i<=sqrt(x);i++)

   if(x%i==0)
    ok=1;
    return ok;
}

int i_prim(int x)
{
    int i,p1,p2,dif;
    for(i=1;i<=x;i++)
    if(prim(i)==0)
   {
       p1=i;
   }

   for(i=x;i<=x*x;i++)
   if(prim(i)==0)
   {
       p2=i;
       break;
   }
   dif=p2-p1;
   return dif;
}

int main()
{
    int n;
    cin>>n;
    cout<<i_prim(n);
    return 0;
}