結果

問題 No.2494 Sum within Components
ユーザー achapi
提出日時 2023-10-06 21:33:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 182 ms / 2,000 ms
コード長 554 bytes
コンパイル時間 3,529 ms
コンパイル使用メモリ 257,136 KB
最終ジャッジ日時 2025-02-17 04:52:07
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using mint = atcoder::modint998244353;
using namespace std;

int main() {
	int N, M;
	cin >> N >> M;
	vector<int> A(N);
	for (int i = 0; i < N; i++){
		cin >> A[i];
	}
	atcoder::dsu uf(N);
	for (int i = 0; i < M; i++){
		int U, V;
		cin >> U >> V;
		U--;
		V--;
		uf.merge(U, V);
	}
	vector<mint> X(N);
	for (auto s : uf.groups()){
		mint sum = 0;
		for (int i : s){
			sum += A[i];
		}
		for (int i : s){
			X[i] = sum;
		}
	}
	mint ans = 1;
	for (mint m : X){
		ans *= m;
	}
	cout << ans.val() << endl;
}
0