結果

問題 No.2494 Sum within Components
コンテスト
ユーザー なえしら
提出日時 2023-10-06 21:52:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 345 ms / 2,000 ms
コード長 918 bytes
コンパイル時間 313 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 105,452 KB
最終ジャッジ日時 2024-07-26 16:01:40
合計ジャッジ時間 3,624 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353
class DSU:
  def __init__(self, n):
    self.par = [v for v in range(n)]
    self.cc_odr = [1] * n
    self.cc_sum = list(map(int, input().split()))
  def root(self, v):
    if self.par[v] != v:
      self.par[v] = self.root(self.par[v])
    return self.par[v]
  def unite(self, u, v):
    u, v = map(self.root, [u, v])
    if u == v: return
    if self.cc_odr[u] > self.cc_odr[v]:
      u, v = v, u
    self.par[u] = v
    self.cc_odr[v] += self.cc_odr[u]
    self.cc_sum[v] += self.cc_sum[u]
    self.cc_sum[v] %= MOD
  def is_same(self, u, v):
    return self.root(u) == self.root(v)
  def cc_order(self, v):
    return self.cc_odr[self.root(v)]
  def cc_total(self, v):
  	return self.cc_sum[self.root(v)]

N, M = map(int, input().split())
G = DSU(N)
for j in range(M):
  U, V = map(int, input().split())
  G.unite(U-1, V-1)
ans = 1
for i in range(N):
  ans *= G.cc_total(i)
  ans %= MOD
print(ans)
0