結果
問題 | No.2494 Sum within Components |
ユーザー |
|
提出日時 | 2023-10-06 23:01:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 345 ms / 2,000 ms |
コード長 | 1,662 bytes |
コンパイル時間 | 353 ms |
コンパイル使用メモリ | 82,244 KB |
実行使用メモリ | 138,664 KB |
最終ジャッジ日時 | 2024-07-26 16:53:39 |
合計ジャッジ時間 | 3,452 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
class UnionFind: def __init__(self, n, w=None): self.par = [-1]*n self.rank = [0]*n self.siz = [1]*n self.cnt = n self.min_node = [i for i in range(n)] self.weight = w if w else [1]*n def root(self, x): if self.par[x] == -1: return x self.par[x] = self.root(self.par[x]) return self.par[x] def issame(self, x, y): return self.root(x) == self.root(y) def unite(self, x, y): px = self.root(x) py = self.root(y) if px == py: return False if self.rank[px] < self.rank[py]: px, py = py, px self.par[py] = px if self.rank[px] == self.rank[py]: self.rank[px] += 1 self.siz[px] += self.siz[py] self.cnt -= 1 self.min_node[px] = min(self.min_node[px], self.min_node[py]) self.weight[px] += self.weight[py] return False def count(self): return self.cnt def min(self, x): return self.min_node[self.root(x)] def getweight(self, x): return self.weight[self.root(x)] def size(self, x): return self.siz[self.root(x)] import sys input = sys.stdin.readline MOD = 998244353 N, M = map(int, input().split()) A = list(map(int, input().split())) UF = UnionFind(N) for _ in range(M): u, v = map(int, input().split()) u-=1 v-=1 UF.unite(u, v) from collections import defaultdict D = defaultdict(int) for i in range(N): r = UF.root(i) D[r] += A[i] D[r] %= MOD ans = 1 for i in range(N): r = UF.root(i) ans *= D[r] ans %= MOD print(ans)