結果
問題 |
No.2494 Sum within Components
|
ユーザー |
|
提出日時 | 2025-07-26 13:38:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 318 ms / 2,000 ms |
コード長 | 1,651 bytes |
コンパイル時間 | 360 ms |
コンパイル使用メモリ | 82,252 KB |
実行使用メモリ | 133,892 KB |
最終ジャッジ日時 | 2025-07-26 13:38:58 |
合計ジャッジ時間 | 4,828 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
## https://yukicoder.me/problems/no/806 MOD = 998244353 class UnionFind: """ UnionFindの基本的な処理を実装したクラス """ def __init__(self, size): self.root = [i for i in range(size)] self.size = [1] * size def get_root(self, v): if v == self.root[v]: return v else: old_root = self.root[v] new_root = self.get_root(old_root) self.root[v] = new_root return new_root def merge(self, u, v): root_u = self.get_root(u) root_v = self.get_root(v) if root_u == root_v: return False if self.size[root_u] >= self.size[root_v]: self.size[root_u] += self.size[root_v] self.root[root_v] = root_u self.root[v] = root_u else: self.size[root_v] += self.size[root_u] self.root[root_u] = root_v self.root[u] = root_v return True def main(): N, M = map(int, input().split()) A = list(map(int, input().split())) edges = [] for _ in range(M): u, v = map(int, input().split()) edges.append((u - 1, v - 1)) uf = UnionFind(N) for u, v in edges: uf.merge(u, v) components = {} for i in range(N): r = uf.get_root(i) if r not in components: components[r] = 0 components[r] += A[i] components[r] %= MOD answer = 1 for i in range(N): r = uf.get_root(i) answer *= components[r] answer %= MOD print(answer) if __name__ == "__main__": main()