Giải thuật: Ta thấy, số số 3 ước là bình phương của 1 số nguyên tố.

Ví dụ: `5^2 = 25` `-` `5` là số nguyên tố, `25` có `3` ước: `1;5;25`

Tương tự: `3^2=9` `-` `9` có 3 ước: `1;3;9`

....

Code cụ thể:

#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
#define ll long long
#define f ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
using namespace std;
bool snt(ll n)
{
    if(n<2) return false;
    if(n==2) return true;
    if(n%2==0) return false;
    for(ll i=3;i<=sqrt(n);i+=2)
    {
        if(n%i==0) return false;
    }
    return true;
}
bool sd(ll a)
{
    ll Sqrt = sqrt(a);
    if(Sqrt*Sqrt==a) return true;
    else return false;
}
int main()
{
    f
    ll a;
    cin>>a;
    if(sd(a)) cout<<"YES";
    else cout<<"NO";

}
//by nsvdlvt