#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
• 実行時間: 162 ms / 2,000 ms
• コード長: 781 bytes
• コンパイル時間: 1,794 ms
• コンパイル使用メモリ: 201,320 KB
• 実行使用メモリ: 8,216 KB
• 最終ジャッジ日時: 2024-07-26 15:54:06
• 合計ジャッジ時間: 3,330 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,812 KB
testcase_01	AC	3 ms; 6,944 KB
testcase_02	AC	2 ms; 6,940 KB
testcase_03	AC	2 ms; 6,940 KB
testcase_04	AC	2 ms; 6,944 KB
testcase_05	AC	2 ms; 6,944 KB
testcase_06	AC	2 ms; 6,944 KB
testcase_07	AC	1 ms; 6,940 KB
testcase_08	AC	2 ms; 6,944 KB
testcase_09	AC	11 ms; 6,940 KB
testcase_10	AC	12 ms; 6,944 KB
testcase_11	AC	5 ms; 6,940 KB
testcase_12	AC	18 ms; 6,940 KB
testcase_13	AC	12 ms; 6,940 KB
testcase_14	AC	135 ms; 7,112 KB
testcase_15	AC	140 ms; 6,940 KB
testcase_16	AC	67 ms; 7,112 KB
testcase_17	AC	78 ms; 8,136 KB
testcase_18	AC	88 ms; 8,136 KB
testcase_19	AC	162 ms; 8,216 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;
}