結果
問題 | No.2365 Present of good number |
ユーザー |
![]() |
提出日時 | 2023-06-30 21:56:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 48 ms / 2,000 ms |
コード長 | 1,526 bytes |
コンパイル時間 | 175 ms |
コンパイル使用メモリ | 81,840 KB |
実行使用メモリ | 57,728 KB |
最終ジャッジ日時 | 2024-07-07 09:38:21 |
合計ジャッジ時間 | 3,079 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 39 |
ソースコード
""" 2冪になると、あとは 2^k -> 3^k -> 2^(2k) みたいになる 他にループするのあるのかな… 無いと決め打って書いてみようかな f(p1*p2) = f(p1) * f(p2) """ import sys from sys import stdin mod = 10**9+7 dic = {} def f(N,k): p = [] a = [] if k == 0: return N if (N,k) in dic: return dic[(N,k)] Nori = N for i in range(2,N+1): if N % i == 0: p.append(i) a.append(0) while N % i == 0: a[-1] += 1 N //= i ans = 1 #print (p,a) for np,na in zip(p,a): # 2冪の処理 if np == 2: if k % 2 == 0: x = (k//2) now = pow(2 , ( na * pow(2,x,mod-1) ) % (mod-1) , mod ) else: x = (k//2) now = pow(3 , ( na * pow(2,x,mod-1) ) % (mod-1) , mod ) ans = ans * now % mod # 3冪の処理 if np == 3: if k % 2 == 0: x = (k//2) now = pow(3 , ( na * pow(2,x,mod-1) ) % (mod-1) , mod ) else: x = (k//2) + 1 now = pow(2 , ( na * pow(2,x,mod-1) ) % (mod-1) , mod ) ans = ans * now % mod # そうでない場合、愚直に処理 if np > 3: now = f( (np+1) ** na , k-1 ) ans = ans * now % mod dic[(Nori,k)] = ans return ans N,K = map(int,stdin.readline().split()) print (f(N,K))