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

#include<iostream>
#include<cmath>
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<k;i++){
        if(n%i==0){
            run=0;
            break;
        }else {
           run=1;
        }
    }
    if(run==0){
        cout<<"NO";
    }else {
        cout<<"YES";
    }
    return 0;
    
}