Зная простые делители числа и их количество, найти все делители числа

n = int(input())
n1 = n
s = {}
for i in range(2, int(n1 ** 0.5) + 1):
    while n % i == 0:
        if i not in s:
            s[i] = 1
        else:
            s[i] += 1
        n //= i
    if n == 1:
        break
print(s)

import math, random
 
def is_prime(n):
    if n == 1:
        return
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True
 
def prime(n):
    if n < 3:
        return n
    x = random.randint(1, n-2)
    y = 1
    i = 0
    stage = 2
    while math.gcd(n, abs(x - y)) == 1:
        if i == stage:
            y = x
            stage *= 2
        x = (x ** 2 - 1) % n
        i += 1
    return math.gcd(n, abs(x - y))
 
n = int(input())
n1 = n
s = {}
while n != 1:
    x = prime(n)
    while not is_prime(x):
        x = prime(x)
    if x not in s:
        s[x] = 1
    else:
        s[x] += 1
    n = n // x
 
summ = 1
for p, k in s.items():
  summ *= (p**(k+1) - 1) // (p - 1)
 
if summ - n1 == n1:
    print("yes")
else:
    print("no")