下面是我写的代码，求教指点
#include<iostream.h>
#include<math.h>
int main()
{
    int a;
    cin>>a;
    for(int i=2;i<=sqrt(a);i++)
    {
        if(a%i==0)
            break;
    }
    if(i>=sqrt(a)+1)
    {
        cout<<"yes"<<endl;
    }
    else
    {
        cout<<"no"<<endl;
    }
    return 0;
}