并查集的优化

#include <iostream>
#include <cstdio>
using namespace std;
#define maxn 20001
int father[maxn];
int m,n,i,x,y,q;

int find(int x){//寻找根节点编号并压缩路经，递归实现
if(father[x]!=x)
father[x]=find(father[x]);//路径压缩，优化的核心
return father[x];
}

void unio(int r1,int r2){ //合并两个集合
father[r2]=r1;
}

int main(){
cin>>n>>m;
for(i=1;i<=n;i++)
father[i]=i;//建立新的集合，其仅有的成员是i
for(i=1;i<=m;i++){
cin>>x>>y;
int r1=find(x);
int r2=find(y);
if(r1!=r2)unio(r1,r2);
}
cin>>q;
for(i=1;i<=q;i++){
cin>>x>>y;
if(find(x)==find(y))
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}


代码：