結果
問題 | No.2494 Sum within Components |
ユーザー | maeshun |
提出日時 | 2023-10-06 21:34:57 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 189 ms / 2,000 ms |
コード長 | 954 bytes |
コンパイル時間 | 3,875 ms |
コンパイル使用メモリ | 232,920 KB |
実行使用メモリ | 18,944 KB |
最終ジャッジ日時 | 2024-07-26 15:50:07 |
合計ジャッジ時間 | 5,754 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; #define rep(i, n) for(int i=0;i<(n);++i) #define rep1(i, n) for(int i=1;i<=(n);i++) #define ll long long using mint = modint998244353; using P = pair<ll,ll>; using lb = long double; using T = tuple<ll, ll, ll>; #ifdef LOCAL # include <debug_print.hpp> # define dbg(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else # define dbg(...) (static_cast<void>(0)) #endif int main() { int n, m; cin >> n >> m; vector<int> a(n); rep(i,n) cin >> a[i]; dsu uf(n); vector<ll> sum(n); rep(i,m){ int u, v; cin >> u >> v; --u;--v; uf.merge(u,v); } auto res = uf.groups(); for(auto vs: res){ for(int v : vs){ sum[uf.leader(v)]+=a[v]; } } mint ans = 1; rep(i,n){ ans *= sum[uf.leader(i)]; } cout << ans.val() << endl; return 0; }