結果

問題 No.2365 Present of good number
コンテスト
ユーザー H20
提出日時 2023-07-04 12:41:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 914 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,036 KB
実行使用メモリ 55,396 KB
最終ジャッジ日時 2024-07-18 08:37:19
合計ジャッジ時間 3,156 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

import collections
def prime_factorize(n):
    a = []
    while n % 2 == 0:
        a.append(2)
        n //= 2
    f = 3
    while f * f <= n:
        if n % f == 0:
            a.append(f)
            n //= f
        else:
            f += 2
    if n != 1:
        a.append(n)
    return a


N,K = map(int,input().split())
P = prime_factorize(N)
C = collections.Counter(P)
mod = 10**9+7
for _ in range(min(30,K)):
    NC = collections.Counter()
    for k,v in C.items():
        for p in prime_factorize(k+1):
            NC[p]+=v
    C = NC
K = K-min(K,30)
if K==0:
    ans = 1
    for k,v in C.items():
        ans=ans*pow(k,v,mod)%mod
    print(ans)
else:
    if K%2==1:
        NC = collections.Counter()
        for k,v in C.items():
            for p in prime_factorize(k+1):
                NC[p]+=v
        C = NC
    K2=K//2
    print(pow(2,C[2]*pow(2,K2,mod-1),mod)*pow(3,C[3]*pow(2,K2,mod-1),mod)%mod)
0