找朋友並查集裸題

找朋友
並查集用來合併兩個不相交的集合
主要找根，需要路徑優化
合併的時候按秩合併

ac代碼

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e4;
int parent[maxn],rank1[maxn];
int m,n,p;
struct frd{
	int x,y;
}a[maxn],b[maxn];


//找根 
int find_root(int l,int parent[]){
	int l_root=l;
	while(parent[l_root]!=-1){
		l_root=parent[l_root];
	}
	return l_root;
}

//合併
int Union(int l ,int r,int parent[],int rank1[]){
	int x_root=find_root(l,parent);
	int y_root=find_root(r,parent);
	if(x_root==y_root) return 0;
	else{
		if(rank1[x_root]>rank1[y_root]){
			parent[y_root]=x_root;
		}
		else if(rank1[x_root]<rank1[y_root]){
			parent[x_root]=y_root;
		}
		else {
				parent[x_root]=y_root;
				rank1[y_root]++;
			
		}
		
	}
	return 1;
} 

//判斷新來的是否是有一個祖先
bool judge(int l ,int r){
	int x_root=find_root(l,parent);
	int y_root=find_root(r,parent);
	if(x_root==y_root) return true;
	return false;
}
 

int main(){
	memset(parent,-1,sizeof(parent));//表示獨立的節點
	memset(rank1,0,sizeof(rank1));//秩，用來優化樹的高度
	cin>>n>>m>>p;
	for(int i=0;i<m;i++){
		cin>>a[i].x>>a[i].y;
		int result=Union(a[i].x,a[i].y,parent,rank1); 
	}
	for(int i=0;i<p;i++){
		cin>>b[i].x>>b[i].y;
		if(judge(b[i].x,b[i].y)) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	
	
}