結果
問題 |
No.3250 最小公倍数
|
ユーザー |
|
提出日時 | 2025-09-01 12:08:06 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 995 bytes |
コンパイル時間 | 247 ms |
コンパイル使用メモリ | 82,036 KB |
実行使用メモリ | 336,612 KB |
最終ジャッジ日時 | 2025-09-01 12:08:36 |
合計ジャッジ時間 | 28,268 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 17 TLE * 2 -- * 2 |
ソースコード
from collections import Counter N = int(input()) A = list(map(int, input().split())) g = [[] for _ in range(N)] for _ in range(N - 1): u, v = map(int, input().split()) u, v = u - 1, v - 1 g[u].append(v) g[v].append(u) def prime_factorize(n: int): primes = [] while n % 2 == 0: primes.append(2) n //= 2 f = 3 while f * f <= n: if n % f == 0: primes.append(f) n //= f else: f += 2 if n != 1: primes.append(n) return primes dp = [Counter(prime_factorize(a)) for a in A] def dfs(u, p): for v in g[u]: if v == p: continue dfs(v, u) for k, v in dp[v].items(): if k in dp[u]: dp[u][k] = max(dp[u][k], v) else: dp[u][k] = v return dp[u] dfs(0, -1) mod = 998244353 for i in range(N): ans = 1 for k, v in dp[i].items(): ans *= k ** v ans %= mod print(ans)