Help me pls, viết bằng python, nhanh trong 5 phút

from math import trunc,sqrt
def uc(n):
    s=1
    for i in range(2,trunc(sqrt(n))+1):
        if n%i==0:
            s+=i
            if i*i!=n:
                s+=n//i
    return s==n
print(uc(int(input())))

\begin{array}{c} \color{#db1616}{\texttt{#Khoadang09}} \end{array}

def _____(_):
    __ = 1 
    for ___ in range(2,int(_**0.5)+1):
        if _%___==0:
            __ += ___
            if ___ != _ // ___ :
                __ += _ // ___ 
    return __ == _ 
print("YES" if _____(int(input("N = "))) else "NO")