#918352 (C++17) No.2494 Sum within Components

• 問題: No.2494 Sum within Components
• ユーザー: Carpenters-Cat
• 提出日時: 2023-10-06 21:41:33
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 168 ms / 2,000 ms
• コード長: 781 bytes
• コンパイル時間: 1,916 ms
• コンパイル使用メモリ: 199,232 KB
• 実行使用メモリ: 8,324 KB
• 最終ジャッジ日時: 2023-10-06 21:41:37
• 合計ジャッジ時間: 3,822 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,468 KB
testcase_01	AC	2 ms; 5,692 KB
testcase_02	AC	2 ms; 5,404 KB
testcase_03	AC	2 ms; 5,600 KB
testcase_04	AC	2 ms; 5,480 KB
testcase_05	AC	2 ms; 5,604 KB
testcase_06	AC	2 ms; 5,456 KB
testcase_07	AC	3 ms; 5,400 KB
testcase_08	AC	2 ms; 5,400 KB
testcase_09	AC	11 ms; 5,676 KB
testcase_10	AC	12 ms; 5,644 KB
testcase_11	AC	5 ms; 5,540 KB
testcase_12	AC	19 ms; 5,632 KB
testcase_13	AC	13 ms; 5,620 KB
testcase_14	AC	139 ms; 7,356 KB
testcase_15	AC	144 ms; 6,744 KB
testcase_16	AC	70 ms; 7,268 KB
testcase_17	AC	79 ms; 8,324 KB
testcase_18	AC	85 ms; 8,072 KB
testcase_19	AC	168 ms; 8,064 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
int par[200020];
using ll = long long;
ll const m = 998244353;
ll A[200020];
ll val[200020];
int siz[200020];
void init(int N) {
	iota(par, par + N, 0);
	fill(siz, siz + N, 1);
	for (int i = 0; i < N; i ++) val[i] = A[i];
}
int root(int v) {
	return par[v] = (v == par[v] ? v : root(par[v]));
}
void merge(int a, int b) {
	a = root(a); b = root(b);
	if (a == b) return;
	if (siz[a] < siz[b]) swap(a, b);
	par[b] = a;
	siz[a] += siz[b];
	val[a] += val[b];
	val[a] %= m;
}
int main() {
	int N, M;
	cin >> N >> M;
	for (int i = 0; i < N; i ++) cin >> A[i];
	init(N);
	while (M--) {
		int a, b;
		cin >> a >> b;
		merge(--a, --b);
	}
	ll ans = 1;
	for (int i = 0; i < N; i ++) {
		ans = (ans * val[root(i)]) % m;
	}
	cout << ans << endl;
}