結果
問題 | No.2497 GCD of LCMs |
ユーザー |
|
提出日時 | 2023-10-09 03:06:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 428 ms / 2,000 ms |
コード長 | 1,654 bytes |
コンパイル時間 | 1,872 ms |
コンパイル使用メモリ | 82,360 KB |
実行使用メモリ | 84,788 KB |
最終ジャッジ日時 | 2024-07-26 18:21:07 |
合計ジャッジ時間 | 4,844 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 |
ソースコード
def factorization(n):arr = []temp = nfor i in range(2, int(-(-n**0.5//1))+1):if temp%i==0:cnt=0while temp%i==0:cnt+=1temp //= iarr.append([i, cnt])if temp!=1:arr.append([temp, 1])if arr==[]:arr.append([n, 1])return arrimport sysinput = sys.stdin.readlineMOD = 998244353N, M = map(int, input().split())A = list(map(int, input().split()))G = [[] for _ in range(N)]for i in range(M):u, v = map(int, input().split())u-=1v-=1G[u].append(v)G[v].append(u)from collections import defaultdictP = set()F = [defaultdict(int) for _ in range(N)]for i in range(N):a = A[i]f = factorization(a)for p, e in f:if p==1:continueP.add(p)F[i][p] = efrom heapq import heappush, heappopINF = 10 ** 15def dijkstra(s, N, p): # (始点, ノード数)dist = [INF for _ in range(N)]hq = [(F[s][p], s)]dist[s] = F[s][p]seen = [False] * N # ノードが確定済みかどうかwhile hq:d, v = heappop(hq) # ノードを pop するif seen[v]:continueseen[v] = Truefor to in G[v]: # ノード v に隣接しているノードに対してif max(dist[v], F[to][p]) < dist[to]:dist[to] = max(dist[v], F[to][p])heappush(hq, (dist[to], to))return distans = [1 for _ in range(N)]for p in P:dist = dijkstra(0, N, p)for i in range(N):ans[i] *= pow(p, dist[i], MOD)ans[i] %= MODfor a in ans:print(a)