Cho số nguyên dương N. Gọi K là số lượng các ước của N. Hãy kiểm tra xem K có phải là số nguyên tố không? Input: Nhập N Output: Đưa ra thông báo: YES nếu k là số nguyên tố, NO trong trường hợp ngược lại Ví dụ: Input Output 6 NO 9 YES (làm bằng lập trình c++ ,mình cần gấp các bạn ạ)

`Code:`

#include<bits/stdc++.h>
using namespace std;
#define fastread() (ios_base:: sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL))

bool snt(long long n){
 if (n < 2) return false;
 for (int i = 2; i <= sqrt(n); i++)
  if (n % i == 0) return false;
 return true;
}

long long n, cnt;

void solve(){
 cin >> n;
 for (long long i = 1; i <= sqrt(n); i++) {
        if (n % i == 0){
            cnt++;
            if (n/i != i) {
                cnt++;
            }
        }
    }
    if (snt(cnt)) cout << "YES\n";
    else cout << "NO\n";
}

int main(){
 fastread();
 solve();
return 0;
}

$\text{#KL0112}$

#include <iostream>
#include <math.h>
using namespace std;
bool nt(int n) {
    for (int i=2; i<=sqrt(n); i++) if (n%i==0) return false;
    return n>=2;
}
int sum(int n) {
    int d=0;
    for (int i=1; i<=sqrt(n); i++) {
        if (i==sqrt(n)) d++;
        else{
            if (n%i==0) d+=2;
        }
    }
    return d;
}
int main() {
    int n;
    cin >> n;
    if (nt(sum(n))) cout << "YES";
    else cout << "NO";
}