結果

問題 No.2494 Sum within Components
ユーザー Haa
提出日時 2023-10-06 21:28:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 183 ms / 2,000 ms
コード長 1,371 bytes
コンパイル時間 1,640 ms
コンパイル使用メモリ 170,968 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2024-07-26 15:44:13
合計ジャッジ時間 3,382 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef pair<ll,ll> P;
typedef vector<ll> VI;
typedef vector<VI> VVI;
#define REP(i,n) for(int i=0;i<(n);i++)
#define ALL(v) v.begin(),v.end()
template<typename T> bool chmax(T& x, const T& y){return (x<y)?(x=y,true):false;};
template<typename T> bool chmin(T& x, const T& y){return (x>y)?(x=y,true):false;};
constexpr ll MOD=998244353;
constexpr ll INF=2e18;

struct Unionfind{
    vector<int> par, siz;
    Unionfind(int N):par(N), siz(N,1){
        for(int i=0;i<N;i++) par[i]=i;
    }
    int root(int x){
        while(par[x]!=x) x=par[x]=par[par[x]];
        return x;
    }
    void unite(int x, int y){
        x=root(x); y=root(y);
        if(x==y) return;
        if(siz[x]<siz[y]) swap(x,y);
        siz[x]+=siz[y];
        par[y]=x;
    }
    bool same(int x, int y){
        return root(x)==root(y);
    }
    int size(int x){
        return siz[root(x)];
    }
};

int main(){
    int n, m; cin >> n >> m;
    VI a(n); REP(i,n) cin >> a[i];
    Unionfind uf(n);
    int u, v;
    REP(i,m){
        cin >> u >> v;
        u--, v--;
        uf.unite(u,v);
    }
    VI sum(n,0);
    REP(i,n){
        int r=uf.root(i);
        sum[r]=(sum[r]+a[i])%MOD;
    }
    ll ans=1;
    REP(i,n){
        int r=uf.root(i);
        ans=ans*sum[r]%MOD;
    }
    cout << ans << endl;
    return 0;
}
0