結果

問題 No.1293 2種類の道路
ユーザー 沙耶花
提出日時 2020-06-06 09:51:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 84 ms / 2,000 ms
コード長 1,201 bytes
コンパイル時間 2,982 ms
コンパイル使用メモリ 203,148 KB
最終ジャッジ日時 2025-01-10 23:17:34
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:50:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   50 |                 scanf("%d %d",&a,&b);
      |                 ~~~~~^~~~~~~~~~~~~~~
main.cpp:57:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   57 |                 scanf("%d %d",&a,&b);
      |                 ~~~~~^~~~~~~~~~~~~~~

ソースコード

diff # 

#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;
		scanf("%d %d",&a,&b);
		a--;b--;
		uf1.unite(a,b);
	}
	
	for(int i=0;i<Mw;i++){
		int a,b;
		scanf("%d %d",&a,&b);
		a--;b--;
		uf2.unite(a,b);
	}
	
	set<pair<int,int>> P;
	long long ans = 0;
	
	for(int x=0;x<N;x++){
		pair<int,int> t = make_pair(uf1.find(x),uf2.find(x));
		if(P.count(t))continue;
		P.insert(t);
		ans += (long long)uf1.get_size(x)*uf2.get_size(x);
	}
	ans-=N;
	
	cout<<ans<<endl;
	
	return 0;
}
0