結果

問題 No.2494 Sum within Components
ユーザー しゅらにるーとに
提出日時 2024-11-18 21:08:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 224 ms / 2,000 ms
コード長 954 bytes
コンパイル時間 3,885 ms
コンパイル使用メモリ 238,560 KB
実行使用メモリ 28,324 KB
最終ジャッジ日時 2024-11-18 21:08:57
合計ジャッジ時間 6,651 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#include <atcoder/all>
#ifdef LOCAL
#include <debug_print.hpp>
#define debug(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__)
#else
#define debug(...) (static_cast<void>(0))
#endif
using namespace atcoder;
using mint=modint998244353;
using namespace std;
using ll=long long;
using ul=unsigned long long;
int dx[8] = {-1, 1, 0, 0, -1, -1, 1, 1};
int dy[8] = {0, 0, -1, 1, -1, 1, -1, 1};
using Graph=vector<vector<int>>;
ll op(ll a,ll b){return a+b;}
ll e(){return 0;}
int main(){
    int N,M;
    cin>>N>>M;
    dsu uf(N);
    vector<int>A(N);
    for(int i=0;i<N;i++)cin>>A[i];
    for(int i=0;i<M;i++){
        int a,b;
        cin>>a>>b;
        a--,b--;
        uf.merge(a,b);
    }
    vector<vector<int>>G=uf.groups();
    map<int,ll>mp;
    for(int i=0;i<G.size();i++){
        for(auto a:G[i])mp[uf.leader(a)]+=A[a];
    }
    mint ans=1;
    for(int i=0;i<N;i++)ans*=mp[uf.leader(i)];
    cout<<ans.val()<<endl;
}
0