判断质数

题目：

对于大于1的数，如果除了1和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。

输入包括一行，为一个整数N（1 < N ≤1000），正是晓萌给出你让你判断的数字。<n≤1000）。< style="box-sizing: border-box;">

输出包括一行，如果晓萌给出的整数N为质数，那么输出YES；如果N不是质数，那么输出NO。

样例输入

3

样例输出

YES

代码实现：

#include <iostream>
int main(){
    using namespace std;
    
    int N;
    
    cin>>N;
    
    int i; 
    int flag = 1;

    for(i=2; i<N/2; i++){
        if(N%i==0){
            flag = 0 ;
            break;
        }
    }
    
    if(flag==1||N==2){
         cout<<"YES"<<endl;
    }else{
         cout<<"NO"<<endl;
    }
   
    return 0;
}