素数求解

#include<iostream>
#include<math.h>
using namespace std;
void main()
{
int i,a,k=1;
cout<<"请输入一个整数:"<<endl;
cin>>a;
if(a==0 || a==1)
cout<<"不是素数! "<<endl;
else if(a==2||a==3)
cout<<" 是素数"<<endl;
else
{
for(i=2;i<=int(sqrt(a));i++)
{
if(a%i==0)
{
cout<<"不是素数!"<<endl;
k=0;
}
           break;
}
if(k==1)
{
cout<<"是素数!"<<endl;
}
}
}