結果

問題 No.847 Divisors of Power
ユーザー mkawa2
提出日時 2019-07-05 22:37:34
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,494 ms / 2,000 ms
コード長 1,004 bytes
コンパイル時間 243 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 72,356 KB
最終ジャッジ日時 2024-10-06 22:36:12
合計ジャッジ時間 4,919 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import *

def f(n, k, m):
    if n == 1:
        print(1)
        exit()
    tei = []
    sis = []
    cnt = 0
    while n % 2 == 0:
        n //= 2
        cnt += 1
    if cnt:
        tei = [2]
        sis = [cnt * k]
    i = 3
    while i ** 2 <= n:
        cnt = 0
        while n % i == 0:
            cnt += 1
            n //= i
        if cnt:
            tei += [i]
            sis += [cnt * k]
        i += 2
    if n > 1:
        tei += [n]
        sis += [k]
    # print(sis)
    # print(tei)
    kos = len(tei)
    hp = []
    a = [0] * kos
    heappush(hp, [1] + a)
    used = set()
    cnt = 0
    while 1:
        y, *a = heappop(hp)
        if y > m: break
        for i in range(kos):
            if a[i] == sis[i]: continue
            ny = y * tei[i]
            na = a[:]
            na[i] += 1
            if ny in used: continue
            heappush(hp, [ny] + na)
            used.add(ny)
        cnt += 1
    print(cnt)

n, k, m = map(int, input().split())
f(n, k, m)
0