結果
問題 | No.2494 Sum within Components |
ユーザー |
![]() |
提出日時 | 2024-11-12 15:04:49 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 63 ms / 2,000 ms |
コード長 | 1,670 bytes |
コンパイル時間 | 2,172 ms |
コンパイル使用メモリ | 202,204 KB |
最終ジャッジ日時 | 2025-02-25 04:00:03 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> using namespace std; using namespace atcoder; using ll = long long; using mint = modint998244353; struct UnionFind { int ngroup, N; vector<int> par, siz; UnionFind(int _N) : N(_N), par(_N), siz(_N){ ngroup = _N; for(int i = 0; i < N; i++) par[i] = i, siz[i] = 1; } int root(int x) { if (par[x] == x) return x; return par[x] = root(par[x]); } int unite(int x, int y) { int rx = root(x), ry = root(y); if (rx == ry) return rx; ngroup--; if (siz[rx] > siz[ry]) swap(rx, ry); par[rx] = ry; siz[ry] += siz[rx]; return ry; } bool same(int x, int y) { int rx = root(x), ry = root(y); return rx == ry; } int size(int x) {return siz[root(x)];} int group_count() {return ngroup;} vector<vector<int>> groups(){ vector<int> rev(N); int nrt=0; for (int i=0; i<N; i++) if (root(i) == i) rev[i] = nrt, nrt++; vector<vector<int>> res(nrt); for (int i=0; i<N; i++) res[rev[root(i)]].push_back(i); return res; } }; int main(){ cin.tie(nullptr); ios_base::sync_with_stdio(false); ll N, M, U, V; cin >> N >> M; vector<ll> A(N); vector<mint> P(N); mint ans=1; for (int i=0; i<N; i++) cin >> A[i]; UnionFind tree(N); for (int i=0; i<M; i++){ cin >> U >> V; U--; V--; tree.unite(U, V); } for (int i=0; i<N; i++){ P[tree.root(i)] += A[i]; } for (int i=0; i<N; i++){ ans *= P[tree.root(i)]; } cout << ans.val() << endl; return 0; }