結果
問題 |
No.2365 Present of good number
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)