判断质数，是输出yes，不是输出no

#include
#include
using namespace std;
int main()
{
   int n,i,run=1,k;
    cin>>n;
    k=sqrt(n)+1;
    if(n<2){
    	cout<<"NO";
    	return 0;
	}
	if(n==2){
		cout<<"YES";
		return 0;
	}
    for(i=2;i