結果
問題 | No.2494 Sum within Components |
ユーザー |
![]() |
提出日時 | 2023-11-24 20:52:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 61 ms / 2,000 ms |
コード長 | 895 bytes |
コンパイル時間 | 2,149 ms |
コンパイル使用メモリ | 205,644 KB |
最終ジャッジ日時 | 2025-02-17 23:38:07 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/dsu>#include <atcoder/modint>using namespace std;using namespace atcoder;using mint = modint998244353;void fast_io() {ios::sync_with_stdio(false);std::cin.tie(nullptr);}int main() {fast_io();int n, m;cin >> n >> m;vector<int> a(n);for (auto &x : a) cin >> x;dsu uf(n);vector<mint> su(n);for (int i = 0; i < n; i++) su[i] = a[i];for (int i = 0; i < m; i++) {int u, v;cin >> u >> v;u--, v--;if (uf.same(u, v)) {continue;}mint x = su[uf.leader(u)];mint y = su[uf.leader(v)];uf.merge(u, v);su[uf.leader(u)] = x + y;}mint ans = 1;for (int i = 0; i < n; i++) {if (uf.leader(i) == i) {ans *= su[i].pow(uf.size(i));}}cout << ans.val() << endl;}