結果

問題 No.2576 LCM Pattern
ユーザー LyricalMaestro
提出日時 2024-01-06 18:32:07
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,567 bytes
コンパイル時間 299 ms
コンパイル使用メモリ 82,484 KB
実行使用メモリ 133,380 KB
最終ジャッジ日時 2024-09-27 19:18:48
合計ジャッジ時間 4,889 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 10 TLE * 1 -- * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

## https://yukicoder.me/problems/no/2576
MOD = 998244353
import math
def calc_gcd(A, B):
"""
A, B
"""
a = max(A, B)
b = min(A, B)
while a % b > 0:
c = a % b
a = b
b = c
return b
def main():
N, M = map(int, input().split())
# M
sqrt_m = int(math.sqrt(M))
divisors = []
for p in range(1, sqrt_m + 1):
if M % p == 0:
q = M // p
divisors.append(p)
if q != p:
divisors.append(q)
divisors.sort()
lcm_map = {}
for d1 in divisors:
for d2 in divisors:
gcd = calc_gcd(d1, d2)
lcm_map[(d1, d2)] = (d1 // gcd) * d2
def solve(map1, map2, divisors):
new_map = {}
for d1 in divisors:
for d2 in divisors:
lcm = lcm_map[(d1, d2)]
if M % lcm != 0:
break
if d1 not in map1 or d2 not in map2:
continue
if lcm not in new_map:
new_map[lcm] = 0
new_map[lcm] += (map1[d1] * map2[d2]) % MOD
new_map[lcm] %= MOD
return new_map
answer_map = {1:1}
base_map = {d: 1 for d in divisors}
while N > 0:
if N % 2 == 1:
answer_map = solve(answer_map, base_map, divisors)
N //= 2
base_map = solve(base_map, base_map, divisors)
print(answer_map[M])
if __name__ == "__main__":
main()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0