結果

問題 No.2912 0次パーシステントホモロジー
ユーザー Carpenters-Cat
提出日時 2024-10-04 22:41:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 262 ms / 2,000 ms
コード長 1,000 bytes
コンパイル時間 2,925 ms
コンパイル使用メモリ 207,208 KB
最終ジャッジ日時 2025-02-24 15:30:47
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
	}
	sort(eg.begin(), eg.end());
	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) {
			gs -= merge(get<1>(eg[fl]), get<2>(eg[fl]));
			fl ++;
		}
		ans[i] = gs;
	}
	for (auto& a : ans) {
		cout << a << endl;
	}
}
0