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 NKế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

làm bằng python nhé mình đang cần gấp:

import math
def getDiv(x):
    sq=int(math.sqrt(x))
    ans=0
    for i in range(1, sq + 1):
        if x % i == 0:
            if i != x // i:
                ans += 1
            ans += 1
    if ans < 2:
        return False
    for i in range(2,int(math.sqrt(ans))+1):
        if ans % i == 0:
            return False
    return True

n = int(input())
if getDiv(n):
    print("YES")
else:
    print("NO")

def snt(n):
    s=0
    #meothinhle
    for i in range(1,int(n**0.5)+1):
        if n%i==0: s+=2
    k=int(n**0.5)
    if k*k==n: s-=1
    if s<2: return "NO"
    for i in range(2,int(s**0.5)+1):
        if s%i==0: return "NO"
    return "YES"
n=int(input())
print(snt(n))