結果

問題 No.3250 最小公倍数
ユーザー i_taku
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0