质数判断（张老师）

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
 
using namespace std;
 
const int N = 1e6+10;
 
int n,m;
int va[N];
 
signed main()
{
    IOS;
    long long n;
    cin>>n;
    int suc = 1;
    if(n==1) suc = 0;
    for(int i=2;i<=n/i;i++)
    {
        if(n%i==0)
        {
            suc = 0;
            break;
        }
    }
    if(suc) cout<<"YES";
    else cout<<"NO";
    return 0;
}