結果

問題 No.416 旅行会社
ユーザー naimonon77
提出日時 2016-08-30 14:48:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 315 ms / 4,000 ms
コード長 2,810 bytes
コンパイル時間 2,125 ms
コンパイル使用メモリ 188,636 KB
実行使用メモリ 14,848 KB
最終ジャッジ日時 2024-12-14 20:14:47
合計ジャッジ時間 6,557 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

// antaさんのやつを勝手に解析しているもの

#include "bits/stdc++.h"
using namespace std;
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define rer(i,l,u) for(int (i)=(int)(l);(i)<=(int)(u);++(i))
#define reu(i,l,u) for(int (i)=(int)(l);(i)<(int)(u);++(i))
static const int INF = 0x3f3f3f3f; static const long long INFL = 0x3f3f3f3f3f3f3f3fLL;
typedef vector<int> vi; typedef pair<int, int> pii; typedef vector<pair<int, int> > vpii; typedef long long ll;
template<typename T, typename U> static void amin(T &x, U y) { if (y < x) x = y; }
template<typename T, typename U> static void amax(T &x, U y) { if (x < y) x = y; }

struct UnionFind {
	vector<int> data;
	void init(int n) { data.assign(n, -1); }
	bool unionSet(int x, int y) {
		x = root(x); y = root(y);
		if (x != y) {
			if (data[y] < data[x]) swap(x, y);
			data[x] += data[y]; data[y] = x;
		}
		return x != y;
	}
	bool findSet(int x, int y) { return root(x) == root(y); }
	int root(int x) { return data[x] < 0 ? x : data[x] = root(data[x]); }
	int size(int x) { return -data[root(x)]; }
};

int main() {
	int N; int M; int Q;
	while (~scanf("%d%d%d", &N, &M, &Q)) {
		set<pii> edges;
		rep(i, M) {
			int A; int B;
			scanf("%d%d", &A, &B), --A, --B;
			if (A > B)swap(A, B);
			// A <= B
			edges.insert(make_pair(A, B));
		}
		// edge[i].first <= edge[i].second
		vpii queries(Q); //動的確保
		rep(i, Q) {
			int A; int B; // <- a,b同じの使うのか
			scanf("%d%d", &A, &B), --A, --B;
			if (A > B)swap(A, B);
			edges.erase(make_pair(A, B));
			queries[i] = make_pair(A, B);
		}
		// この時点でedgeは破壊されない橋の集合である
		// queries[i].first <= queries[i].second
		UnionFind uf;
		uf.init(N);
		for (auto e : edges)
			uf.unionSet(e.first, e.second);
		vector<int> ans(N, 0);
		// 絶対に破壊されない橋だけを繋いでも
		// 0とつながっていると言うことは
		// 橋が全て壊されたあともその島に到達できる : -1
		rep(i, N) if (uf.findSet(i, 0))
			ans[i] = -1;
		vector<vi> as(N);
		// as[i] : iが持っている子の列
		rep(i, N)
			as[uf.root(i)].push_back(i);
		
		for (int i = Q - 1; i >= 0; --i) {
			// vpii queries(Q);
			auto e = queries[i];
			int x = e.first, y = e.second;
			x = uf.root(x), y = uf.root(y);
			if (x == y) continue;
			// 根が同じなら以下の作業は不要
			if (uf.findSet(0, y))
				swap(x, y);
			// yは0を含まないroot

			if (uf.findSet(0, x)) {
				for (int j : as[y])
					ans[j] = i + 1;
			}
			if (as[x].size() < as[y].size())
				as[x].swap(as[y]);
			// as[x].size() >= as[y].size()
			as[x].insert(as[x].end(), as[y].begin(), as[y].end());
			as[y].clear();
			uf.unionSet(x, y);
			as[uf.root(x)].swap(as[x]);
		}
		for (int i = 1; i < N; ++i)
			printf("%d\n", ans[i]);
	}
	return 0;
}
0