結果
問題 | No.2494 Sum within Components |
ユーザー |
|
提出日時 | 2023-10-27 16:10:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 943 ms / 2,000 ms |
コード長 | 756 bytes |
コンパイル時間 | 229 ms |
コンパイル使用メモリ | 82,188 KB |
実行使用メモリ | 174,224 KB |
最終ジャッジ日時 | 2024-09-25 13:04:22 |
合計ジャッジ時間 | 6,085 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
import sys sys.setrecursionlimit(10**6) N, M = map(int, input().split()) A = list(map(int, input().split())) UV = [list(map(int, input().split())) for _ in range(M)] MOD = 998244353 G = [[] for _ in range(N)] for i in range(M): u, v = UV[i] u -= 1 v -= 1 G[u].append(v) G[v].append(u) link = [-1] * N def dfs(now): for nxt in G[now]: if link[now] == link[nxt]: continue link[nxt] = link[now] dfs(nxt) cnt = 0 for i in range(N): if link[i] != -1: continue cnt += 1 link[i] = cnt dfs(i) a = [[0] * 2 for _ in range(cnt)] for i in range(N): a[link[i]-1][0] += A[i] a[link[i]-1][1] += 1 ans = 1 for x, y in a: ans *= x ** y ans %= MOD print(ans)