結果

問題 No.2494 Sum within Components
コンテスト
ユーザー inkyaman
提出日時 2024-03-24 16:46:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 213 ms / 2,000 ms
コード長 1,367 bytes
コンパイル時間 1,068 ms
コンパイル使用メモリ 122,524 KB
最終ジャッジ日時 2025-02-20 13:40:24
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <math.h>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <numeric>
#include <iomanip>
#include <climits>
#include <functional>
#include <cassert>
#include <tuple>
using namespace std;
using ll = long long;

int N,M;
vector<int> G[202020];
int A[202020];
ll mod = 998244353;

int main() {

    cin >> N >> M;
    for(int i = 1; i <= N; ++i) cin >> A[i];
    for(int i = 0; i < M; ++i) {
        int u,v; cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    ll ans = 1;
    vector<bool> vis(N+1);
    vector<ll> S(N+1);
    for(int i = 1; i <= N; ++i) {
        if(vis[i]) continue;
        ll sum = 0;
        vis[i] = true;
        queue<int> q;
        vector<int> route;
        q.push(i);
        route.push_back(i);
        while(!q.empty()) {
            int v = q.front(); q.pop();
            sum += A[v];
            sum %= mod;
            for(int ne : G[v]) {
                if(vis[ne]) continue;
                vis[ne] = true;
                q.push(ne);
                route.push_back(ne);
            }
        }

        for(int v : route) S[v] = sum;

    }

    for(int i = 1; i <= N; ++i) {
        ans *= S[i];
        ans %= mod;
    }

    cout << ans << endl;

}
0