#1015439 (C++17) No.2912 ０次パーシステントホモロジー

• 問題: No.2912 ０次パーシステントホモロジー
• ユーザー: Carpenters-Cat
• 提出日時: 2024-10-04 22:38:45
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 971 bytes
• コンパイル時間: 1,981 ms
• コンパイル使用メモリ: 210,700 KB
• 実行使用メモリ: 6,400 KB
• 最終ジャッジ日時: 2024-10-04 22:38:50
• 合計ジャッジ時間: 4,185 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	2 ms; 5,248 KB
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	AC	1 ms; 5,248 KB
testcase_08	AC	1 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	1 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	WA	-
testcase_13	WA	-
testcase_14	AC	2 ms; 5,248 KB
testcase_15	AC	19 ms; 5,248 KB
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-

ソースコード

#include <bits/stdc++.h>
using namespace std;
int par[200020];
int siz[200020];
int gs;
void init(int N) {
	gs = N;
	iota(par, par + N, 0);
	fill(siz, siz + N, 1);
}
int root(int v) {
	return par[v] = (v == par[v] ? v : root(par[v]));
}
int merge(int a, int b) {
	a = root(a); b = root(b);
	if (a == b) return 0;
	if (siz[a] < siz[b]) swap(a, b);
	par[b] = a;
	siz[a] += siz[b];
	return 1;
}
int main () {
	int N, M;
	cin >> N >> M;
	init(N);
	std::vector<tuple<int, int, int>> eg(M);
	for (auto& [w, u, v] : eg) {
		cin >> u >> v >> w;
	}
	int Q;
	cin >> Q;
	int fl = 0;
	vector<int> R(Q);
	for (auto& r : R) cin >> r;
	vector<int> ID(Q);
	iota(ID.begin(), ID.end(), 0);
	sort(ID.begin(), ID.end(), [&](int i, int j) {return R[i] < R[j];});
	vector<int> ans(Q);
	for (auto& i : ID) {
		int r = R[i];
		while (fl < M && get<0>(eg[fl]) <= r) {
			fl ++;
			gs -= merge(get<1>(eg[fl]), get<2>(eg[fl]));
		}
		ans[i] = gs;
	}
	for (auto& a : ans) {
		cout << a << endl;
	}
}