結果
問題 |
No.2494 Sum within Components
|
ユーザー |
|
提出日時 | 2023-10-06 21:48:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 246 ms / 2,000 ms |
コード長 | 1,068 bytes |
コンパイル時間 | 3,915 ms |
コンパイル使用メモリ | 257,604 KB |
最終ジャッジ日時 | 2025-02-17 04:58:52 |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; using ll=long long; using mint=modint998244353; int main() { int n,m; cin>>n>>m; vector<mint> a(n); for(int i=0;i<n;i++){ int t; cin>>t; a[i]=t; } vector<vector<int>> graph(n); for(int i=0;i<m;i++){ int u,v; cin>>u>>v; u--;v--; graph[u].push_back(v); graph[v].push_back(u); } vector<mint> sum; vector<vector<int>> hist; vector<bool> seen(n); function<void(int,int)> rec=[&](int i,int t)->void{ seen[i]=true; sum[t]+=a[i]; hist[t].push_back(i); for(int x:graph[i]){ if(!seen[x])rec(x,t); } }; int cnt=0; for(int i=0;i<n;i++){ if(seen[i])continue; cnt++; sum.push_back(0); hist.push_back(vector<int>()); rec(i,cnt-1); } mint ans=1; for(int i=0;i<cnt;i++){ for(int j=0;j<hist[i].size();j++)ans*=sum[i]; } cout<<ans.val()<<endl; }