#475454 (C++17) No.1293 2種類の道路

• 問題: No.1293 2種類の道路
• ユーザー: 沙耶花
• 提出日時: 2020-05-02 16:07:15
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 131 ms / 2,000 ms
• コード長: 1,238 bytes
• コンパイル時間: 2,066 ms
• コンパイル使用メモリ: 215,008 KB
• 実行使用メモリ: 15,188 KB
• 最終ジャッジ日時: 2024-07-04 18:31:35
• 合計ジャッジ時間: 4,534 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	126 ms; 11,856 KB
testcase_10	AC	128 ms; 11,860 KB
testcase_11	AC	124 ms; 11,784 KB
testcase_12	AC	123 ms; 11,720 KB
testcase_13	AC	131 ms; 11,864 KB
testcase_14	AC	82 ms; 15,188 KB
testcase_15	AC	80 ms; 15,060 KB
testcase_16	AC	112 ms; 13,144 KB
testcase_17	AC	114 ms; 13,780 KB
testcase_18	AC	72 ms; 14,160 KB
testcase_19	AC	69 ms; 15,060 KB
testcase_20	AC	72 ms; 15,188 KB
testcase_21	AC	76 ms; 5,376 KB
testcase_22	AC	76 ms; 5,376 KB
testcase_23	AC	78 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define modulo 1000000007
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 10000000000000000

struct unionfind{
	vector<int> data;
	vector<int> size;
	unionfind(int n){
		for(int i=0;i<n;i++){
			data.push_back(i);
			size.push_back(1);
		}
	}
	
	int find(int x){
		if(data[x]==x)return x;
		return data[x]=find(data[x]);
	}
	
	bool unite(int x,int y){
		x=find(x);y=find(y);
		if(x==y)return false;
		if(size[x]>size[y])swap(x,y);
		data[x]=y;
		size[y]+=size[x];
		return true;
	}
	
	bool check(int x,int y){
		return (find(x)==find(y));
	}
	
	int get_size(int x){
		int X = find(x);
		return size[X];
	}
};

int main(){
	
	int N,Md,Mw;
	cin>>N>>Md>>Mw;
	
	unionfind uf1(N),uf2(N);
	
	for(int i=0;i<Md;i++){
		int a,b;
		cin>>a>>b;
		a--;b--;
		uf1.unite(a,b);
	}
	
	for(int i=0;i<Mw;i++){
		int a,b;
		cin>>a>>b;
		a--;b--;
		uf2.unite(a,b);
	}
	
	vector<set<int>> S(N);
	
	for(int i=0;i<N;i++){
		S[uf1.find(i)].insert(uf2.find(i));
	}
	
	vector<long long> cnt(N,0);
	for(int i=0;i<N;i++){
		for(auto a:S[i]){
			cnt[i] += (long long)uf2.get_size(a);
		}
	}
	
	long long ans = 0;
	
	for(int i=0;i<N;i++){
		ans += cnt[uf1.find(i)]-1;
	}
	cout<<ans<<endl;
	
	return 0;
}