結果
| 問題 |
No.2365 Present of good number
|
| コンテスト | |
| ユーザー |
👑 SPD_9X2
|
| 提出日時 | 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))
SPD_9X2