結果

問題 No.1293 2種類の道路
コンテスト
ユーザー mamimume____mo
提出日時 2020-11-21 16:39:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 137 ms / 2,000 ms
コード長 1,225 bytes
コンパイル時間 1,774 ms
コンパイル使用メモリ 177,988 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-07-23 15:45:41
合計ジャッジ時間 4,522 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
class UnionFind {
public:
	vector <int> par;
	vector <int> siz;
	UnionFind(int sz_): par(sz_), siz(sz_, 1) {
		for (int i = 0; i < sz_; ++i) par[i] = i;
	}
	void init(int sz_) {
		par.resize(sz_);
		siz.assign(sz_, 1);
		for (int i = 0; i < sz_; ++i) par[i] = i;
	}
	int root(int x) {
		while (par[x] != x) {
			x = par[x] = par[par[x]];
		}
		return x;
	}
	bool merge(int x, int y) {
		x = root(x);
		y = root(y);
		if (x == y) return false;
		if (siz[x] < siz[y]) swap(x, y);
		siz[x] += siz[y];
		par[y] = x;
		return true;
	}
	bool issame(int x, int y) {
		return root(x) == root(y);
	}
	int size(int x) {
		return siz[root(x)];
	}
};


int main() {
	int N,D,W;
	cin >> N >> D >> W;
	UnionFind uf_d(N);
	UnionFind uf_w(N);
	for(int i = 0;i < D;i++){
		int a,b;
		cin >> a >> b;
		a--,b--;
		uf_d.merge(a,b);
	}
	for(int i = 0;i < W;i++){
		int c,d;
		cin >> c >> d;
		c--,d--;
		uf_w.merge(c,d);
	}
	ll ans = 0;
	map<pair<int,int>,bool> used;
	for(int i = 0;i < N;i++){
		if(!used[make_pair(uf_d.root(i),uf_w.root(i))]){
			ans += (ll)uf_d.size(i) * uf_w.size(i);
		}
		used[make_pair(uf_d.root(i),uf_w.root(i))] = true;
	}

	cout << ans - N << endl;
}
0