Số nguyên tố (Tên file: Ngto1.*) 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? Dữ liệu: Nhập N Kết quả: Đư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ụ: Dữ liệu Kết quả 6 NO 9 YES Giải thích: 6 có 4 ước là 1, 2, 3, 6. Khi đó 4 không là số nguyê tố 9 có 3 ước là 1, 3, 9. Khi đó 3 là số nguyên tố.

#include <iostream>
using namespace std;
bool is_prime(int num) {
    if (num <= 1) 
        return false;
    if (num <= 3) 
        return true;
    if (num % 2 == 0 || num % 3 == 0) 
        return false;
    for (int i = 5; i * i <= num; i += 6) {
        if (num % i == 0 || num % (i + 2) == 0) {
            return false;
        }
    }
    return true;
}

int count_divisors(int num) {
    int count = 0;
    for (int i = 1; i <= num; ++i) {
        if (num % i == 0) {
            ++count;
        }
    }
    return count;
}

int main() {
    int N;
    cin >> N;
    int K = count_divisors(N);
    if (is_prime(K)) {
        cout << "YES" << endl;
    } else {
        cout << "NO" << endl;
    }
    return 0;
}

def check(n):
    if n<2:
        return False
    for i in range(2,int(n**0.5)+1):
        if n%i == 0:
            return False
    return True

def demuoc(n):
    d = 0
    for i in range(1,int(n**0.5)+1):
        if n%i == 0:
            d += 1
            if n/i != i:
                d += 1
    return d

n = int(input())
if check(demuoc(n)):
    print('YES')
else:
    print('NO')