并查集（Union-Find Set）模板

By Stockholm

并查集（Union-Find Set）模板

题目描述
如题，现在有一个并查集，你需要完成合并和查询操作。
输入输出格式
输入格式：
第一行包含两个整数 N、M，表示共有 N 个元素和 M 个操作。
接下来 M 行，每行包含三个整数 Zi、Xi、Yi
当 Zi=1 时，将 Xi 与 Yi 所在的集合合并
当 Zi=2 时，输出 Xi 与 Yi 是否在同一集合内，是的话输出 Y；否则
话输出 N
输出格式：
如上，对于每一个 Zi=2 的操作，都有一行输出，每行包含一个大写
字母，为 Y 或者 N
输入输出样例输入样例#1：
4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4
输出样例#1：
N Y N Y
说明
时空限制：1000ms,128M
数据规模：
对于 30%的数据，N<=10，M<=20；对于 70%的数据，N<=100，M<=1000；
对于 100%的数据，N<=10000，M<=200000。

思路

思路是这样的，需要将每一个点对其列出一个祖先数组，初始化任何
一个节点都为本节点的祖先节点，然后在合并集合时，注意进行路径
压缩，边寻找边压缩路径，以便下一次的判断。操作 2 就简单一些，
只需要两个节点在更新寻找节点的同时进行对比并更新即可。

代码（C++）

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;int i,j,n,m,t;int b[10001];int r(){int aans=0;char ch=getchar();while(ch<'0'||ch>'9')ch=getchar();while(ch>='0'&&ch<='9'){aans*=10;aans+=ch-'0';ch=getchar();}return aans;}int update(int xx){return xx==b[xx]?xx:b[xx]=update(b[xx]);}int main(){n=r(),m=r();int x,y,z;for(i=1;i<=n;i++)b[i]=i;for(i=1;i<=m;i++){z=r(),x=r(),y=r();if(z==1){b[update(x)]=update(y);}if(z==2){if(update(x)==update(y))cout<<"Y"<<endl;elsecout<<"N"<<endl;}}return 0;}